Home
last modified time | relevance | path

Searched defs:out_depth (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h41 int out_depth; member
Ddeep_conv2d.h81 int out_depth; member
Ddeep_conv2d.cc50 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
Dconv_ops_3d.cc136 const int64 out_depth = filter.dim_size(4); in Compute() local
222 const int64 out_depth = filter.dim_size(4); in launch() local
Ddepthwise_conv_op.cc89 const int64 out_depth = args.out_depth; in Run() local
327 const int32 out_depth = in_depth * depth_multiplier; in Compute() local
Deigen_spatial_convolutions_test.cc678 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
Dconv_ops.cc174 int out_cols, int out_depth, int dilation_rows, in Run()
215 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
229 int out_cols, int out_depth, int dilation_rows, in Run()
367 const int out_depth = static_cast<int>(filter.dim_size(3)); in ComputeConv2DDimension() local
Dconv_ops_test.cc1105 int filter_w, int filter_h, int out_depth) { in Conv2D()
1129 int filter_h, int out_depth) { in Conv2DWithBias()
1155 int out_depth) { in Conv2DWithBiasAndRelu()
1176 int out_depth) { in Conv2DWithBatchNorm()
1211 int out_depth) { in Conv2DWithBatchNormAndRelu()
1232 int out_depth, in FusedConv2DWithBias()
1265 int out_depth, const std::vector<string>& fused_ops = {}) { in FusedConv2DWithBatchNorm()
Dpooling_ops_common.h66 int out_depth; member
Ddepthwise_conv_grad_op.cc286 const int64 out_depth = args.out_depth; in ComputeBackpropInput() local
754 const int64 out_depth = args.out_depth; in ComputeBackpropFilter() local
901 const int64 out_depth = args.out_depth; in operator ()() local
Dfractional_avg_pool_op.cc247 const int64 out_depth = out_backprop.dim_size(3); in Compute() local
Ddepthwise_conv_op_gpu.h92 const int out_depth = args.out_depth; in DepthwiseConv2dGPUKernelNHWC() local
332 const int out_depth = args.out_depth; in DepthwiseConv2dGPUKernelNCHW() local
Dmkl_conv_ops.h373 int out_depth; variable
Dconv_ops_using_gemm.cc486 const int out_depth = static_cast<int>(filter.dim_size(3)); in Compute() local
Dquantized_conv_ops.cc520 const int64 out_depth = filter.dim_size(3); in Compute() local
Dnn_ops_test.cc106 int out_depth, int filter_rows, int filter_cols, in BM_ConvFloat()
500 int out_depth, int filter_rows, in BM_ConvFloatDepthwise()
Dconv_grad_filter_ops.cc141 auto out_depth = output.dimension(3); in operator ()() local
Dconv_grad_input_ops.cc148 auto out_depth = output_backward.dimension(3); in operator ()() local
Dconv_ops_fused_image_transform.cc789 const int out_depth = static_cast<int>(filter.dim_size(3)); in Compute() local
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/
Dgraph_functions_wrapper.c216 uint32_t* out_width, uint32_t* out_depth, uint8_t* out_vals, in hexagon_controller_ExecuteGraph()
259 uint32_t out_batches, out_height, out_width, out_depth; in hexagon_controller_ExecuteInceptionDummyData() local
Dhexagon_controller.c187 const uint32_t out_depth = output0->depth; in hexagon_controller_ExecuteGraphWithBuffer() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc86 const int32 out_depth = in_depth * depth_multiplier; in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc73 int out_depth = 5; in TEST() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py368 self, batch, input_shape, filter_shape, in_depth, out_depth, stride, argument
/external/mesa3d/src/mesa/state_tracker/
Dst_cb_drawpixels.c125 struct ureg_dst out_color, out_depth, out_stencil; in get_drawpix_z_stencil_program() local

12