| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | resize_nearest_neighbor_test.cc | 32 const RuntimeShape& output_shape, in TestReferenceResizeNearestNeighbor()
52 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
63 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
75 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
87 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
98 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
109 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
121 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
133 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
144 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
[all …]
|
| D | transpose_utils_test.cc | 25 RuntimeShape output_shape({9}); in TEST() local
43 RuntimeShape output_shape({3, 9}); in TEST() local
63 RuntimeShape output_shape({1, 9}); in TEST() local
82 RuntimeShape output_shape({8, 4, 3}); in TEST() local
104 RuntimeShape output_shape({8, 4, 1}); in TEST() local
126 RuntimeShape output_shape({1, 4, 1}); in TEST() local
146 RuntimeShape output_shape({1, 1, 1}); in TEST() local
166 RuntimeShape output_shape({3, 9, 4, 2}); in TEST() local
190 RuntimeShape output_shape({3, 9, 4, 1}); in TEST() local
213 RuntimeShape output_shape({3, 1, 4, 1}); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_test.cc | 40 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
61 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
82 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
103 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
123 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
143 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
214 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); in TEST() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | WinogradConvolutionLayerFixture.h | 59 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
88 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, const Pa… in compute_target()
129 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_reference()
163 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
197 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, const Pa… in compute_target()
328 …TensorShape output_shape = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_… in setup() local
354 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
387 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
410 …TensorShape output_shape = compute_winograd_filter_transform_shape(TensorInfo(input_shape, 1, dat… in setup() local
436 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
[all …]
|
| D | SpaceToBatchFixture.h | 43 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
63 … const TensorShape &block_shape_shape, const TensorShape &paddings_shape, TensorShape output_shape, in compute_target()
116 … const TensorShape &output_shape, DataType data_type, QuantizationInfo quantization_info) in compute_reference()
144 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
156 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
|
| D | FullyConnectedLayerFixture.h | 58 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, bool tra… in setup()
103 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, bool… in compute_target()
193 …, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape) in compute_reference()
221 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, bool tra… in setup()
235 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, bool tra… in setup()
|
| D | ReshapeLayerFixture.h | 48 void setup(TensorShape input_shape, TensorShape output_shape, DataType data_type) in setup()
61 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, DataTyp… in compute_target()
94 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Data… in compute_reference()
|
| D | CopyFixture.h | 47 void setup(TensorShape input_shape, TensorShape output_shape, DataType data_type) in setup()
61 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, DataTyp… in compute_target()
94 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Data… in compute_reference()
|
| D | DeconvolutionLayerFixture.h | 53 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
126 …ape input_shape, TensorShape weights_shape, const TensorShape bias_shape, TensorShape output_shape, in compute_target()
185 …, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, in compute_reference()
233 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
254 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
275 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
|
| D | PriorBoxLayerFixture.h | 52 …const TensorShape output_shape = misc::shape_calculator::compute_prior_box_shape(input_info, info); in setup() local
60 …TensorType compute_target(TensorShape input_shape, TensorShape output_shape, PriorBoxLayerInfo inf… in compute_target()
95 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Prio… in compute_reference()
|
| D | Im2ColFixture.h | 65 …const TensorShape output_shape = compute_im2col_conv_shape(&input_info, _kernel_dims, _conv_info, … in setup() local
78 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, DataTyp… in compute_target()
107 …void compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, DataType d… in compute_reference()
|
| D | DepthToSpaceLayerFixture.h | 43 …void setup(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, DataType data_t… in setup()
56 … TensorType compute_target(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, in compute_target()
93 const TensorShape &output_shape, DataType data_type) in compute_reference()
|
| D | SpaceToDepthFixture.h | 43 …void setup(TensorShape input_shape, TensorShape output_shape, const int block_shape, DataType data… in setup()
56 …TensorType compute_target(TensorShape input_shape, TensorShape output_shape, const int block_shape, in compute_target()
91 … SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, in compute_reference()
|
| D | Col2ImFixture.h | 56 …const TensorShape output_shape = compute_col2im_shape(TensorInfo(input_shape, 1, data_type), convo… in setup() local
69 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, const S… in compute_target()
98 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, unsi… in compute_reference()
|
| D | WeightsReshapeFixture.h | 54 …const TensorShape output_shape = compute_weights_reshaped_shape(TensorInfo(input_shape, 1, data_ty… in setup() local
67 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, const b… in compute_target()
108 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
|
| D | DirectConvolutionLayerFixture.h | 69 … const TensorShape output_shape = compute_deep_convolution_shape(input_info, weights_info, info); in setup() local
76 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
128 …hape, TensorShape weights_shape, const TensorShape &bias_shape, TensorShape output_shape, const Pa… in compute_target()
175 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_reference()
228 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
241 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
|
| D | DirectConvolutionLayerTensorShiftFixture.h | 61 const TensorShape output_shape = get_output_shape(input_shape, weights_shape, info); in setup() local
69 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
113 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_target()
164 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_reference()
234 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
247 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
|
| D | BatchToSpaceLayerFixture.h | 43 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape output_shape, DataT… in setup()
56 …pe compute_target(TensorShape input_shape, TensorShape block_shape_shape, TensorShape output_shape, in compute_target()
104 const TensorShape &output_shape, DataType data_type) in compute_reference()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util_test.cc | 38 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST() local
48 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST() local
57 Shape output_shape = ShapeUtil::MakeShape(F32, {20, 7}); in TEST() local
69 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST() local
81 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST() local
95 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 16, 7}); in TEST() local
108 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 15, 2, 14}); in TEST() local
121 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 1, 7}); in TEST() local
134 Shape output_shape = ShapeUtil::MakeShape(F32, {8, 16, 1}); in TEST() local
154 Shape output_shape = ShapeUtil::MakeShape(F32, {}); in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | sub.h | 37 const RuntimeShape& output_shape, in SubNonBroadcast()
53 const RuntimeShape& output_shape, in SubNonBroadcast()
74 const RuntimeShape& output_shape, in BroadcastSubSlow()
114 const RuntimeShape& output_shape, in BroadcastSubSlow()
171 const RuntimeShape& output_shape, in BroadcastSubSlow()
211 const RuntimeShape& output_shape, in BroadcastSubSlow()
265 const RuntimeShape& output_shape, int64_t* output_data) { in BroadcastSubSlow()
302 const RuntimeShape& output_shape, T* output_data) { in BroadcastSubSlow()
341 const RuntimeShape& output_shape, in BroadcastSub16POTSlow()
449 const RuntimeShape& output_shape, uint8_t* output_data) { in Sub()
[all …]
|
| D | pad.h | 40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl()
120 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
130 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
141 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
151 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle()
160 const RuntimeShape& output_shape, in PadImageStyle()
|
| D | reference_ops.h | 94 const RuntimeShape& output_shape, T* output_data) { in Relu()
106 const RuntimeShape& output_shape, T* output_data) { in Relu1()
119 const RuntimeShape& output_shape, float* output_data) { in Relu6()
134 const RuntimeShape& output_shape, T* output_data) { in ReluX()
152 const RuntimeShape& output_shape, T* output_data) { in ReluX()
175 const RuntimeShape& output_shape, in BroadcastMulFivefold()
225 const RuntimeShape& output_shape, int16* output_data) { in Mul()
244 const RuntimeShape& output_shape, uint8* output_data) { in Mul()
273 const RuntimeShape& output_shape, int16_t* output_data) { in Sub16()
320 const Scalar* const* input_data, const RuntimeShape& output_shape, in Pack()
[all …]
|
| D | logistic.h | 31 const RuntimeShape& output_shape, float* output_data) { in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
70 const RuntimeShape& output_shape, int16_t* output_data) { in Logistic()
95 const RuntimeShape& output_shape, int8_t* output_data, in Logistic()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 43 xla::Shape output_shape = in GetOutputShape() local
55 const xla::Shape& output_shape) { in CompileGrad()
77 auto output_shape = GetOutputShape(ctx); in CompileForward() local
131 auto output_shape = in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense.cc | 40 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
53 const TfLiteTensor* output_shape, in CheckDimensionsMatch()
129 const TfLiteTensor* output_shape, in ResizeOutputShape()
149 const TfLiteTensor* output_shape; in Prepare() local
205 const TfLiteTensor* output_shape; in SparseToDenseImpl() local
|