Searched defs:in_sizes (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dindex_ops_kernel_argmax_float_2d.cc32 int64* in_sizes = static_cast<int64*>(data[1]); local
36 Eigen::DSizes<Eigen::DenseIndex, 2> in_eig_sizes(in_sizes[0], in_sizes[1]);
/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_tfconv_op.h154 size_t* in_sizes = new size_t[ndims]; local
158 in_sizes[i] = input_shape.GetSizes()[i];
171 delete[] in_sizes;
H A Dmkl_conv_grad_bias_ops.cc96 mkl_context.in_sizes[MklDims::W] =
98 mkl_context.in_sizes[MklDims::H] =
100 mkl_context.in_sizes[MklDims::C] =
102 mkl_context.in_sizes[MklDims::N] =
105 mkl_context.in_sizes[MklDims::W] = GetTensorDim(input, data_format_, 'W');
106 mkl_context.in_sizes[MklDims::H] = GetTensorDim(input, data_format_, 'H');
107 mkl_context.in_sizes[MklDims::C] = GetTensorDim(input, data_format_, 'C');
108 mkl_context.in_sizes[MklDims::N] = GetTensorDim(input, data_format_, 'N');
110 mkl_context.in_sizes);
119 mkl_context.in_dims, mkl_context.in_sizes),
146 size_t in_sizes[4]; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon26517
[all...]
H A Dmkl_aggregate_ops.cc106 mkl_context.in_sizes = new size_t[mkl_context.in_dims];
114 mkl_context.in_sizes[i] = tmp_mkl_shape->GetSizes()[i];
119 mkl_context.in_sizes[i] =
125 mkl_context.in_strides[i - 1] * mkl_context.in_sizes[i - 1];
145 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.in_strides);
179 size_t* in_sizes = nullptr; member in struct:tensorflow::MklAddNOp::__anon26512
190 CHECK_EQ(dnnLayoutCreate_F32(&lt_input1, in_dims, in_sizes, in_strides),
198 CHECK_EQ(dnnLayoutCreate_F32(&lt_input2, in_dims, in_sizes, in_strides),
275 delete[] in_sizes;
H A Dmkl_conv_grad_input_ops.cc157 mkl_context.in_sizes[0] =
159 mkl_context.in_sizes[1] =
161 mkl_context.in_sizes[2] = static_cast<size_t>(dims.in_depth);
162 mkl_context.in_sizes[3] = static_cast<size_t>(dims.batch_size);
182 mkl_context.in_sizes);
200 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
210 mklOutputShape.SetTfLayout(mkl_context.in_dims, mkl_context.in_sizes,
239 size_t in_sizes[4]; member in struct:tensorflow::MklConv2DCustomBackpropInputOp::__anon26519
H A Dmkl_lrn_op.cc169 size_t in_sizes[4]; member in struct:tensorflow::MklLRNOp::__anon26524
403 mkl_context.in_sizes[i] = mkl_context.out_sizes[i] =
497 size_t in_sizes[4]; member in struct:tensorflow::MklLRNGradOp::__anon26525
516 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides),
H A Dmkl_pooling_ops_common.h405 size_t in_sizes[4]; member in struct:tensorflow::__anon26528
H A Dmkl_conv_grad_filter_ops.cc159 mkl_context.in_sizes[0] =
161 mkl_context.in_sizes[1] =
163 mkl_context.in_sizes[2] = static_cast<size_t>(backprop_dims.in_depth);
164 mkl_context.in_sizes[3] = static_cast<size_t>(backprop_dims.batch_size);
179 mkl_context.in_sizes);
210 mkl_context.in_sizes, mkl_context.out_sizes,
270 size_t in_sizes[4]; member in struct:tensorflow::MklConv2DCustomBackpropFilterOp::__anon26518
305 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides),
H A Dmkl_conv_ops.cc213 mkl_context.in_sizes[MklDims::W] = static_cast<size_t>(input_cols);
214 mkl_context.in_sizes[MklDims::H] = static_cast<size_t>(input_rows);
215 mkl_context.in_sizes[MklDims::C] = static_cast<size_t>(input_depth);
216 mkl_context.in_sizes[MklDims::N] = static_cast<size_t>(batch);
232 mkl_context.in_sizes);
260 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
268 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.out_sizes,
334 size_t in_sizes[4]; member in struct:tensorflow::MklConv2DOp::__anon26520
357 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides),
H A Dmkl_relu_op.cc93 mkl_context.in_sizes = new size_t[mkl_context.in_dims];
96 mkl_context.in_sizes[i] = mkl_context.input_shape.GetSizes()[i];
101 mkl_context.in_sizes = new size_t[mkl_context.in_dims];
104 mkl_context.in_sizes[i] = input.dim_size((mkl_context.in_dims - 1) - i);
109 mkl_context.in_strides[i - 1] * mkl_context.in_sizes[i - 1];
127 mkl_context.in_dims, mkl_context.in_sizes, mkl_context.in_strides);
154 size_t* in_sizes; member in struct:tensorflow::MklReluOp::__anon26529
165 free(in_sizes);
174 CHECK_EQ(dnnLayoutCreate_F32(&lt_input, in_dims, in_sizes, in_strides),
195 size_t* in_sizes; member in struct:tensorflow::MklReluGradOp::__anon26530
[all...]
H A Dmkl_fused_batch_norm_op.cc135 mkl_context.mkl_params.in_sizes,
198 size_t orig_size = mkl_context.mkl_params.in_sizes[0] *
199 mkl_context.mkl_params.in_sizes[1] *
200 mkl_context.mkl_params.in_sizes[3];
220 size_t in_sizes[4]; member in struct:tensorflow::MklFusedBatchNormOp::__anon26521::MklBatchNormParams
247 mkl_params.in_sizes[0] = static_cast<size_t>(
250 mkl_params.in_sizes[1] = static_cast<size_t>(
253 mkl_params.in_sizes[2] = static_cast<size_t>(
256 mkl_params.in_sizes[3] = static_cast<size_t>(
259 mkl_params.depth = mkl_params.in_sizes[
498 size_t in_sizes[4]; member in struct:tensorflow::MklFusedBatchNormGradOp::__anon26522::MklBatchNormParams
[all...]

Completed in 182 milliseconds