| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 163 const int64_t in_depth = input.dim_size(3); in operator ()() local
273 const int64_t in_depth = input.dim_size(3); in operator ()() local
337 const int64_t in_depth = GetTensorDim(input, data_format, 'C'); in operator ()() local
368 int input_cols, int in_depth, int filter_rows, in Run()
384 int input_cols, int in_depth, int filter_rows, in Run()
425 int input_cols, int in_depth, int filter_rows, in Run()
439 int input_cols, int in_depth, int filter_rows, in Run()
571 const int in_depth = static_cast<int>(in_depth_raw); in ComputeConv2DDimension() local
|
| D | conv_ops_benchmark_test.cc | 84 static Conv2DGraph Conv2D(int batch, int height, int width, int in_depth, in Conv2D()
119 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBias()
145 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBiasAndActivation()
168 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBatchNorm()
205 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBatchNormAndActivation()
229 int in_depth, int filter_w, int filter_h, in FusedConv2DWithBias()
273 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in FusedConv2DWithBatchNorm()
|
| D | deep_conv2d.cc | 50 int in_depth, int out_depth, int out_rows, in GetDeepConvCost()
75 static int64_t GetDirectConvCost(int filter_rows, int filter_cols, int in_depth, in GetDirectConvCost()
99 int filter_cols, int in_depth, int out_depth, in CanUseDeepConv2D()
207 const int64_t in_depth = args.in_depth; in operator ()() local
387 const int64_t in_depth = args.in_depth; in operator ()() local
542 const int64_t in_depth = args.in_depth; in operator ()() local
904 const int64_t in_depth = args.in_depth; in operator ()() local
957 const int64_t in_depth = args.in_depth; in operator ()() local
|
| D | depthwise_conv_op.h | 30 int in_depth; member
|
| D | deep_conv2d.h | 72 int in_depth; member
|
| D | conv_grad_shape_utils.h | 53 int64_t in_depth, out_depth; member
|
| D | conv_ops_3d.cc | 148 const int64_t in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
244 const int64_t in_depth = GetTensorDim(input, data_format, 'C'); in launch() local
|
| D | eigen_spatial_convolutions_test.cc | 669 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
|
| D | conv_grad_input_ops_benchmark_test.cc | 40 int in_depth, int filter_h, int filter_w, in Conv2DBackpropInput()
|
| D | conv_grad_filter_ops_benchmark_test.cc | 41 int in_depth, int filter_h, int filter_w, in Conv2DBackpropFilter()
|
| D | fractional_avg_pool_op.cc | 299 const int64_t in_depth = orig_input_tensor_shape_flat(3); in Compute() local
|
| D | depthwise_conv_op.cc | 364 const int64_t in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
|
| D | conv_ops_using_gemm.cc | 481 const int64_t in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
|
| D | quantized_conv_ops.cc | 538 const int64_t in_depth = input.dim_size(3); in Compute() local
|
| D | fused_batch_norm_op.cc | 1301 const int64_t in_depth = GetTensorDim(x, tensor_format_, 'C'); in ComputeWithReservedSpace() local
1546 const int64_t in_depth = GetTensorDim(x, tensor_format_, 'C'); in ComputeWithReservedSpace() local
|
| D | nn_ops_test.cc | 110 int rows, int cols, int in_depth, int out_depth, in BM_ConvFloat()
519 int rows, int cols, int in_depth, in BM_ConvFloatDepthwise()
|
| D | conv_grad_filter_ops.cc | 218 auto in_depth = input.dimension(3); in operator ()() local
|
| D | conv_ops_fused_image_transform.cc | 780 const int64_t in_depth = padded_shape.dim_size(3); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 214 in_depth = input_shape.dimensions(feature_dim); in MakeXlaForwardConvOp() local
308 int64_t in_depth = input_shape.dimensions(feature_dim), in MakeXlaBackpropInputConvOp() local
432 int64_t in_depth = input_shape.dimensions(c_dim), in MakeXlaBackpropFilterConvOp() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | use_cudnn.cc | 118 const int32_t in_depth, in ShouldCudnnGroupedConvolutionBeUsed()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | padding.h | 87 int in_height, int in_width, int in_depth, int filter_height, in ComputePadding3DValues()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 72 int in_depth = 3; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 33 int in_depth, int single_buffer_length, in ExtractPatchIntoBufferColumn()
292 int in_depth, int in_height, int in_width, int in_channel, // Input shape. in ExtractPatchIntoBufferColumn3D()
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | conv_ops_3d_test.py | 471 self, batch, input_shape, filter_shape, in_depth, out_depth, stride, argument
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 232 const int32_t in_depth = kernel_shape[2]; in DepthwiseConv2dNative() local
585 const int32_t in_depth = kernel_shape[2]; in Im2Row() local
|