| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | category_encoding.py | 192 def sparse_bincount(inputs, out_depth, binary_output, count_weights=None): argument
210 def dense_bincount(inputs, out_depth, binary_output, count_weights=None): argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | use_cudnn.cc | 75 const int32 out_depth) { in IsCudnnSupportedFilterSize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_benchmark_test.cc | 77 int filter_w, int filter_h, int out_depth, in Conv2D()
109 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBias()
135 int out_depth, const string& activation_type, in Conv2DWithBiasAndActivation()
158 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBatchNorm()
195 int out_depth, const string& activation_type, in Conv2DWithBatchNormAndActivation()
219 int out_depth, in FusedConv2DWithBias()
259 int out_depth, const std::vector<string>& fused_ops = {}, in FusedConv2DWithBatchNorm()
|
| D | deep_conv2d.cc | 50 int out_depth, int out_rows, int out_cols) { in GetDeepConvCost()
75 int out_depth, int out_rows, int out_cols) { in GetDirectConvCost()
98 int filter_cols, int in_depth, int out_depth, in CanUseDeepConv2D()
382 const int64 out_depth = args.out_depth; in operator ()() local
532 const int64 out_depth = args.out_depth; in operator ()() local
891 const int64 out_depth = args.out_depth; in operator ()() local
944 const int64 out_depth = args.out_depth; in operator ()() local
|
| D | depthwise_conv_op.h | 41 int out_depth; member
|
| D | deep_conv2d.h | 81 int out_depth; member
|
| D | conv_grad_shape_utils.h | 53 int64 in_depth, out_depth; member
|
| D | depthwise_conv_op.cc | 94 const int64 out_depth = args.out_depth; in Run() local
353 const int32 out_depth = in_depth * depth_multiplier; in Compute() local
|
| D | conv_ops_3d.cc | 143 const int64 out_depth = filter.dim_size(4); in Compute() local
232 const int64 out_depth = filter.dim_size(4); in launch() local
|
| D | conv_ops.cc | 241 int out_cols, int out_depth, int dilation_rows, in Run()
282 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
296 int out_cols, int out_depth, int dilation_rows, in Run()
434 const int out_depth = static_cast<int>(filter.dim_size(3)); in ComputeConv2DDimension() 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_filter_ops_benchmark_test.cc | 42 int out_depth, int stride_h, int stride_w, in Conv2DBackpropFilter()
|
| D | conv_grad_input_ops_benchmark_test.cc | 41 int out_depth, int stride_h, int stride_w, in Conv2DBackpropInput()
|
| D | depthwise_conv_grad_op.cc | 298 const int64 out_depth = args.out_depth; in ComputeBackpropInput() local
795 const int64 out_depth = args.out_depth; in ComputeBackpropFilter() local
942 const int64 out_depth = args.out_depth; in operator ()() local
|
| D | pooling_ops_common.h | 74 int out_depth; member
|
| D | fractional_avg_pool_op.cc | 247 const int64 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 | 520 const int64 out_depth = filter.dim_size(3); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 86 const int32 out_depth = in_depth * depth_multiplier; in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv3d.cc | 94 int out_width, out_height, out_depth; in Prepare() local
|
| D | padding.h | 85 int* out_width, int* out_depth) { in ComputePadding3DValues()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 73 int out_depth = 5; in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 379 int out_depth; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 214 out_depth = filter_shape.dimensions(attrs.num_spatial_dims + 1), in MakeXlaForwardConvOp() 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
|