Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc34 const int output_depth = 5; in TEST() local
83 const int output_depth = 5; in TEST() local
142 const int output_depth = 5; in TEST() local
205 const int output_depth = 5; in TEST() local
261 const int output_depth = 3; in TEST() local
309 const int output_depth = 3; in TEST() local
367 const int output_depth = 5; in TEST() local
438 const int output_depth = 5; in TEST() local
503 const int output_depth = 3; in TEST() local
558 const int output_depth = 3; in TEST() local
[all …]
Ddepthtospace_op.cc100 const int output_depth = input_depth / block_size_sq; in Compute() local
156 const int output_depth = output.dimension(3); in operator ()() local
Dspacetodepth_op_gpu.cu.cc37 const int output_width, const int output_depth, in S2D_NHWC()
153 const int output_depth = output.dimension(3); in operator ()() local
179 const int output_depth = output.dimension(1); in operator ()() local
Ddepthtospace_op_gpu.cu.cc39 const int output_width, const int output_depth, in D2S_NHWC()
157 const int output_depth = output.dimension(3); in operator ()() local
185 const int output_depth = output.dimension(1); in operator ()() local
Deigen_spatial_convolutions_test.cc33 const int output_depth = 10; in TEST() local
78 const int output_depth = 10; in TEST() local
198 const int output_depth = 7; in TEST() local
250 const int output_depth = 7; in TEST() local
308 const int output_depth = 7; in TEST() local
364 const int output_depth = 7; in TEST() local
412 const int output_depth = 6; in TEST() local
460 const int output_depth = 7; in TEST() local
511 const int output_depth = 7; in TEST() local
562 const int output_depth = 7; in TEST() local
[all …]
Dspacetodepth_op.cc101 const int output_depth = input_depth * block_size_ * block_size_; in Compute() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_following_affine.cc79 const int output_depth = weights_shape.dims(0); in FuseAddOrSubParamsIntoFollowingAffine() local
95 const int output_depth = in FuseAddOrSubParamsIntoFollowingAffine() local
Dfuse_binary_into_preceding_affine.cc136 int output_depth; in FuseMulOrDivParamsIntoPrecedingAffine() local
Dpropagate_fixed_sizes.cc34 void ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, in ComputeConvSizes()
144 const int output_depth = weights_shape.dims(0); in ProcessConvOperator() local
272 const int output_depth = weights_shape.dims(3); in ProcessDepthwiseConvOperator() local
891 const int output_depth = input_shape.dims(3); in ProcessAveragePoolOperator() local
908 const int output_depth = input_shape.dims(3); in ProcessMaxPoolOperator() local
927 const int output_depth = input_shape.dims(3); in ProcessL2PoolOperator() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h45 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnected() local
91 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
139 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
193 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected() local
Ddepthwiseconv_float.h45 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConv() local
Ddepthwiseconv_uint8.h119 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in Run() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmean.h39 const int output_depth = output_shape.Dims(3); in Mean() local
Dfully_connected.h43 const int output_depth = output_shape.Dims(1); in FullyConnected() local
Dconv.h52 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
Ddepthwise_conv.h51 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h155 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtest_util.cc23 bool ComputeConvSizes(const RuntimeShape& input_shape, int output_depth, in ComputeConvSizes()
Ddepthwiseconv_float_test.cc79 const int output_depth = input_depth * depth_multiplier; in TryTestOneDepthwiseConv() local
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc230 int output_depth) { in ShuffleAndSetWeights()
549 TfLiteRegistration* registration, int input_depth, int output_depth, in SimpleTestQuantizedInt16OutputCase()
657 int output_depth = 4 * output_depth_numblocks; in TEST_P() local
Ddepthwise_conv_test.cc93 int output_depth = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv_test.cc67 int output_depth = filter_dims->data[3]; in TestDepthwiseConvFloat() local
147 int output_depth = filter_dims->data[3]; in TestDepthwiseConvQuantized() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc119 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvOptimizedForFilterWidthEight() local
343 const int output_depth = GetTensorShape(filter).Dims(3); in EvalQuantized() local
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc205 const int32 output_depth = GetFilterDim(filter, filter_format_, 'O'); in Compute() local
478 const int output_depth = GetFilterDim(filter_param, filter_format, 'O'); in launch() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc692 DimensionHandle output_depth; in DepthwiseConv2DNativeShape() local
924 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolShape() local
1026 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolV2Shape() local

12