Home
last modified time | relevance | path

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

123

/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
135 const int output_depth = 3; in TEST() local
183 const int output_depth = 3; in TEST() local
235 const int output_depth = 3; in TEST() local
290 const int output_depth = 3; in TEST() local
342 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid()
402 int output_depth = 1; in TEST() local
418 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
479 int output_depth = 1; in TEST() local
[all …]
Deigen_backward_cuboid_convolutions_test.cc41 const int output_depth = 5; in TEST() local
104 const int output_depth = 5; in TEST() local
167 const int output_depth = 5; in TEST() local
238 const int output_depth = 5; in TEST() local
310 const int output_depth = 5; in TEST() local
377 const int output_depth = 5; in TEST() local
437 const int output_depth = 5; in TEST() local
503 const int output_depth = 5; in TEST() local
570 const int output_depth = 3; in TEST() local
637 const int output_depth = 3; in TEST() local
[all …]
Ddepthtospace_op_gpu.cu.cc38 const int output_width, const int output_depth, in D2S_NHWC()
156 const int output_depth = output.dimension(3); in operator ()() local
184 const int output_depth = output.dimension(1); 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.cc100 const int output_depth = input_depth / block_size_sq; in Compute() local
155 const int output_depth = output.dimension(3); in operator ()() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup.cc59 const auto output_depth = OutputDepth(weight_dims); in SetWeightSubmatrixDims() local
91 const auto output_depth = OutputDepth(weight_dims); in DecomposeQuantLstmWeightsTensor() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h42 const int output_depth = output_shape.Dims(1); in FullyConnected() local
82 const int output_depth = output_shape.Dims(1); in FullyConnected() local
Dmean.h40 const int output_depth = output_shape.Dims(3); in Mean() local
Ddepthwise_conv.h50 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
147 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
230 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvHybridPerChannel() local
Dconv.h52 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
155 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
Dtranspose_conv.h44 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
143 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dconv_per_channel_quantized_16x8_test.cc60 const int output_depth = output_shape.Dims(3); in PickOutputMultiplier() local
116 int output_activation_max, int output_depth, in PickReasonableMultiplier()
155 const int output_depth = 8 * ExponentialRandomPositiveInt(0.9f, 10, 50); in GenerateValidShapeConfigurations() local
212 const int output_depth = output_shape_inference.Dims(3); in TryTestOneConvFilter() local
Daveragepool_quantized_test.cc65 const int output_depth = input_depth; in CreateDataAndRunAveragePool() local
128 const int output_depth = input_depth; in CreateExtremalDataAndRunAveragePool() local
Ddepthwiseconv_per_channel_quantized_16x8_test.cc115 int output_activation_max, int output_depth, in PickReasonableMultiplier()
155 const int output_depth = input_depth * depth_multiplier; in GenerateValidShapeConfigurations() local
211 const int output_depth = output_shape_inference.Dims(3); in TryTestOneDepthwiseConv3x3Filter() local
Ddepthwiseconv_per_channel_quantized_test.cc116 int output_activation_max, int output_depth, in PickReasonableMultiplier()
203 const int output_depth = input_depth * depth_multiplier; in GenerateValidShapeConfigurations() local
256 const int output_depth = output_shape_inference.Dims(3); in TryTestOneDepthwiseConv3x3Filter() local
Dmaxpool_quantized_test.cc61 const int output_depth = input_depth; in CreateDataAndRunMaxPool() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h42 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnected() local
87 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
134 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
187 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected() local
Dconv.h47 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
130 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
210 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in HybridConvPerChannel() local
Dspace_to_depth.h41 const int output_depth = output_shape.Dims(3); in SpaceToDepth() local
Ddepth_to_space.h41 const int output_depth = output_shape.Dims(3); in DepthToSpace() local
Dtranspose_conv.h43 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
133 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
Ddepthwiseconv_uint8.h118 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in Run() local
205 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in RunPerChannel() 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
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h45 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight() local
94 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight1x4Impl() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h45 const int output_depth = in TransposeConvV2() local

123