| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_benchmark_test.cc | 85 int filter_w, int filter_h, int out_depth, in Conv2D()
120 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBias()
146 int out_depth, const string& activation_type, in Conv2DWithBiasAndActivation()
169 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBatchNorm()
206 int out_depth, const string& activation_type, in Conv2DWithBatchNormAndActivation()
230 int out_depth, in FusedConv2DWithBias()
274 int out_depth, const std::vector<string>& fused_ops = {}, in FusedConv2DWithBatchNorm()
|
| D | deep_conv2d.cc | 50 int in_depth, int out_depth, int out_rows, in GetDeepConvCost()
76 int out_depth, int out_rows, int out_cols) { in GetDirectConvCost()
99 int filter_cols, int in_depth, int out_depth, in CanUseDeepConv2D()
388 const int64_t out_depth = args.out_depth; in operator ()() local
543 const int64_t out_depth = args.out_depth; in operator ()() local
905 const int64_t out_depth = args.out_depth; in operator ()() local
958 const int64_t out_depth = args.out_depth; in operator ()() local
|
| D | depthwise_conv_op.h | 41 int out_depth; member
|
| D | conv_ops.cc | 274 const int64_t out_depth = output->dim_size(3); in operator ()() local
386 int out_cols, int out_depth, int dilation_rows, in Run()
427 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
441 int out_cols, int out_depth, int dilation_rows, in Run()
582 const int out_depth = static_cast<int>(filter.dim_size(3)); in ComputeConv2DDimension() local
|
| D | deep_conv2d.h | 81 int out_depth; member
|
| D | conv_grad_shape_utils.h | 53 int64_t in_depth, out_depth; member
|
| D | depthwise_conv_op.cc | 96 const int64_t out_depth = args.out_depth; in Run() local
374 const int32_t out_depth = in_depth * depth_multiplier; in Compute() local
|
| D | conv_ops_3d.cc | 152 const int64_t out_depth = filter.dim_size(4); in Compute() local
250 const int64_t out_depth = filter.dim_size(4); in launch() local
|
| D | eigen_spatial_convolutions_test.cc | 678 const int out_depth = in_depth; in TEST() local
739 const int out_depth = in_depth; in TEST() local
801 const int out_depth = 3; in TEST() local
853 const int out_depth = 3; in TEST() local
907 const int out_depth = in_depth; in TEST() local
973 const int out_depth = in_depth; in TEST() local
1039 const int out_depth = 3; in TEST() local
1094 const int out_depth = 3; in TEST() local
1151 const int out_depth = ceil_div(in_depth, stride); in TEST() local
1221 const int out_depth = ceil_div(in_depth, stride); in TEST() local
|
| D | conv_grad_input_ops_benchmark_test.cc | 41 int out_depth, int stride_h, int stride_w, in Conv2DBackpropInput()
|
| D | conv_grad_filter_ops_benchmark_test.cc | 42 int out_depth, int stride_h, int stride_w, in Conv2DBackpropFilter()
|
| D | depthwise_conv_grad_op.cc | 301 const int64_t out_depth = args.out_depth; in ComputeBackpropInput() local
799 const int64_t out_depth = args.out_depth; in ComputeBackpropFilter() local
946 const int64_t out_depth = args.out_depth; in operator ()() local
|
| D | pooling_ops_common.h | 74 int out_depth; member
|
| D | fractional_avg_pool_op.cc | 277 const int64_t out_depth = out_backprop.dim_size(3); in Compute() local
|
| D | conv_ops_using_gemm.cc | 488 const int out_depth = static_cast<int>(filter.dim_size(3)); in Compute() local
|
| D | quantized_conv_ops.cc | 545 const int64_t out_depth = filter.dim_size(3); in Compute() local
|
| D | nn_ops_test.cc | 110 int rows, int cols, int in_depth, int out_depth, in BM_ConvFloat()
520 int depth_multiplier, int out_depth, in BM_ConvFloatDepthwise()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | use_cudnn.cc | 119 const int32_t out_depth) { in ShouldCudnnGroupedConvolutionBeUsed()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 405 int out_depth; variable
505 int out_depth = GetTensorDim(out_shape, data_format_, 'C'); variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | padding.h | 89 int* out_width, int* out_depth) { in ComputePadding3DValues()
|
| D | conv3d.cc | 169 int out_width, out_height, out_depth; in Prepare() local
|
| D | pooling3d.cc | 317 int out_width, out_height, out_depth; in GenericPrepare() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 73 int out_depth = 5; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 213 out_depth = filter_shape.dimensions(attrs.num_spatial_dims + 1), in MakeXlaForwardConvOp() local
|
| /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
|