| /external/tensorflow/tensorflow/core/util/ |
| D | use_cudnn.cc | 74 const int32 filter_cols, const int32 in_depth, in IsCudnnSupportedFilterSize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 158 const int64 in_depth = GetTensorDim(input, data_format, 'C'); in operator ()() local
196 const int64 in_depth = GetTensorDim(input, data_format, 'C'); in operator ()() local
223 int input_cols, int in_depth, int filter_rows, in Run()
239 int input_cols, int in_depth, int filter_rows, in Run()
280 int input_cols, int in_depth, int filter_rows, in Run()
294 int input_cols, int in_depth, int filter_rows, in Run()
426 const int in_depth = static_cast<int>(in_depth_raw); in ComputeConv2DDimension() local
|
| D | conv_ops_benchmark_test.cc | 76 static Conv2DGraph Conv2D(int batch, int height, int width, int in_depth, in Conv2D()
108 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBias()
134 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBiasAndActivation()
157 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBatchNorm()
194 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in Conv2DWithBatchNormAndActivation()
218 int in_depth, int filter_w, int filter_h, in FusedConv2DWithBias()
258 int batch, int height, int width, int in_depth, int filter_w, int filter_h, in FusedConv2DWithBatchNorm()
|
| D | deep_conv2d.cc | 49 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
|
| 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 in_depth, out_depth; member
|
| D | conv_ops_3d.cc | 139 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
226 const int64 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 | 256 const int64 in_depth = orig_input_tensor_shape_flat(3); in Compute() local
|
| D | depthwise_conv_op.cc | 343 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
|
| D | conv_ops_using_gemm.cc | 481 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute() local
|
| D | quantized_conv_ops.cc | 513 const int64 in_depth = input.dim_size(3); in Compute() local
|
| D | fused_batch_norm_op.cc | 1275 const int64 in_depth = GetTensorDim(x, tensor_format_, 'C'); in ComputeWithReservedSpace() local
1436 const int64 in_depth = GetTensorDim(x, tensor_format_, 'C'); in ComputeWithReservedSpace() local
|
| D | nn_ops_test.cc | 107 int rows, int cols, int in_depth, int out_depth, in BM_ConvFloat()
515 int rows, int cols, int in_depth, in BM_ConvFloatDepthwise()
|
| D | conv_grad_filter_ops.cc | 214 auto in_depth = input.dimension(3); in operator ()() local
|
| D | conv_ops_fused_image_transform.cc | 779 const int64 in_depth = padded_shape.dim_size(3); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 215 in_depth = input_shape.dimensions(feature_dim); in MakeXlaForwardConvOp() local
308 int64 in_depth = input_shape.dimensions(feature_dim), in MakeXlaBackpropInputConvOp() local
432 int64 in_depth = input_shape.dimensions(c_dim), in MakeXlaBackpropFilterConvOp() local
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 73 const int32 in_depth = input.dim_size(3); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | padding.h | 83 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/python/kernel_tests/ |
| D | conv_ops_3d_test.py | 464 self, batch, input_shape, filter_shape, in_depth, out_depth, stride, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 32 int in_depth, int single_buffer_length, in ExtractPatchIntoBufferColumn()
|