Home
last modified time | relevance | path

Searched defs:in_depth (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops.cc141 const int64 in_depth = GetTensorDim(input, data_format, 'C'); in operator ()() local
156 int input_cols, int in_depth, int filter_rows, in Run()
172 int input_cols, int in_depth, int filter_rows, in Run()
213 int input_cols, int in_depth, int filter_rows, in Run()
227 int input_cols, int in_depth, int filter_rows, in Run()
359 const int in_depth = static_cast<int>(in_depth_raw); in ComputeConv2DDimension() local
Ddeep_conv2d.cc49 int out_tile_rows, int out_tile_cols, int in_depth, in GetDeepConvCost()
74 static int64 GetDirectConvCost(int filter_rows, int filter_cols, int in_depth, in GetDirectConvCost()
98 int filter_cols, int in_depth, int out_depth, in CanUseDeepConv2D()
205 const int64 in_depth = args.in_depth; in operator ()() local
381 const int64 in_depth = args.in_depth; in operator ()() local
531 const int64 in_depth = args.in_depth; in operator ()() local
890 const int64 in_depth = args.in_depth; in operator ()() local
943 const int64 in_depth = args.in_depth; in operator ()() local
Ddepthwise_conv_op.h30 int in_depth; member
Ddeep_conv2d.h72 int in_depth; member
Dconv_ops_3d.cc133 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
217 const int64 in_depth = GetTensorDim(input, data_format, 'C'); in launch() local
Ddepthwise_conv_op_gpu.h80 const int in_depth = args.in_depth; in DepthwiseConv2dGPUKernelNHWC() local
187 const int in_depth = args.in_depth; in DepthwiseConv2dGPUKernelNHWCSmall() local
320 const int in_depth = args.in_depth; in DepthwiseConv2dGPUKernelNCHW() local
471 const int in_depth = args.in_depth; in DepthwiseConv2dGPUKernelNCHWSmall() local
Deigen_spatial_convolutions_test.cc669 const int in_depth = 5; in TEST() local
730 const int in_depth = 5; in TEST() local
792 const int in_depth = 5; in TEST() local
844 const int in_depth = 5; in TEST() local
898 const int in_depth = 5; in TEST() local
964 const int in_depth = 5; in TEST() local
1030 const int in_depth = 8; in TEST() local
1085 const int in_depth = 8; in TEST() local
1141 const int in_depth = 8; in TEST() local
1211 const int in_depth = 8; in TEST() local
Dconv_ops_test.cc1104 static Conv2DGraph Conv2D(int batch, int height, int width, int in_depth, in Conv2D()
1128 int in_depth, int filter_w, in Conv2DWithBias()
1152 int width, int in_depth, in Conv2DWithBiasAndRelu()
1174 int width, int in_depth, in Conv2DWithBatchNorm()
1210 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBatchNormAndRelu()
1231 int in_depth, int filter_w, int filter_h, in FusedConv2DWithBias()
1264 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in FusedConv2DWithBatchNorm()
Dfractional_avg_pool_op.cc256 const int64 in_depth = orig_input_tensor_shape_flat(3); in Compute() local
Ddepthwise_conv_op.cc317 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
Dconv_ops_using_gemm.cc479 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
Dquantized_conv_ops.cc513 const int64 in_depth = input.dim_size(3); in Compute() local
Dnn_ops_test.cc105 static void BM_ConvFloat(int iters, int batch, int rows, int cols, int in_depth, in BM_ConvFloat()
499 int in_depth, int depth_multiplier, in BM_ConvFloatDepthwise()
Dconv_grad_filter_ops.cc140 auto in_depth = input.dimension(3); in operator ()() local
Dconv_grad_input_ops.cc147 auto in_depth = input_backward.dimension(3); in operator ()() local
Dconv_ops_fused_image_transform.cc782 const int64 in_depth = padded_shape.dim_size(3); in Compute() local
Ddepthwise_conv_grad_op.cc284 const int64 in_depth = args.in_depth; in ComputeBackpropInput() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc73 const int32 in_depth = input.dim_size(3); in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc72 int in_depth = 3; in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc272 int64 in_depth = filter_shape.dimensions(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py368 self, batch, input_shape, filter_shape, in_depth, out_depth, stride, argument
Dconv_ops_test.py1639 filter_cols, in_depth, out_depth, stride_rows, argument
/external/libpng/contrib/libtests/
Dpngvalid.c2612 static double abserr(const png_modifier *pm, int in_depth, int out_depth) in abserr()
2624 static double calcerr(const png_modifier *pm, int in_depth, int out_depth) in calcerr()
2637 static double pcerr(const png_modifier *pm, int in_depth, int out_depth) in pcerr()
2660 static double outerr(const png_modifier *pm, int in_depth, int out_depth) in outerr()
2692 static double outlog(const png_modifier *pm, int in_depth, int out_depth) in outlog()
2730 static int output_quantization_factor(const png_modifier *pm, int in_depth, in output_quantization_factor()
9427 int in_depth, int out_depth) in init_validate_info()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h451 int in_width, int in_height, int in_depth, int single_buffer_length, in ExtractPatchIntoBufferColumn()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1841 int in_depth, int single_buffer_length, in ExtractPatchIntoBufferColumn()
1931 int in_width, int in_height, int in_depth, int single_buffer_length, in ExtractPatchIntoBufferColumn()

12