/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | index_ops_kernel_argmax_float_2d.cc | 32 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 D | mkl_tfconv_op.h | 154 size_t* in_sizes = new size_t[ndims]; local 158 in_sizes[i] = input_shape.GetSizes()[i]; 171 delete[] in_sizes;
|
H A D | mkl_conv_grad_bias_ops.cc | 96 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 D | mkl_aggregate_ops.cc | 106 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(<_input1, in_dims, in_sizes, in_strides), 198 CHECK_EQ(dnnLayoutCreate_F32(<_input2, in_dims, in_sizes, in_strides), 275 delete[] in_sizes;
|
H A D | mkl_conv_grad_input_ops.cc | 157 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 D | mkl_lrn_op.cc | 169 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(<_input, in_dims, in_sizes, in_strides),
|
H A D | mkl_pooling_ops_common.h | 405 size_t in_sizes[4]; member in struct:tensorflow::__anon26528
|
H A D | mkl_conv_grad_filter_ops.cc | 159 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(<_input, in_dims, in_sizes, in_strides),
|
H A D | mkl_conv_ops.cc | 213 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(<_input, in_dims, in_sizes, in_strides),
|
H A D | mkl_relu_op.cc | 93 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(<_input, in_dims, in_sizes, in_strides), 195 size_t* in_sizes; member in struct:tensorflow::MklReluGradOp::__anon26530 [all...] |
H A D | mkl_fused_batch_norm_op.cc | 135 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...] |