Home
last modified time | relevance | path

Searched defs:input_shape (Results 1 – 25 of 593) sorted by relevance

12345678910>>...24

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtranspose_utils_test.cc24 RuntimeShape input_shape({9}); in TEST() local
42 RuntimeShape input_shape({9, 3}); in TEST() local
62 RuntimeShape input_shape({9, 1}); in TEST() local
81 RuntimeShape input_shape({4, 3, 8}); in TEST() local
103 RuntimeShape input_shape({4, 1, 8}); in TEST() local
125 RuntimeShape input_shape({4, 1, 1}); in TEST() local
145 RuntimeShape input_shape({1, 1, 1}); in TEST() local
165 RuntimeShape input_shape({9, 3, 2, 4}); in TEST() local
189 RuntimeShape input_shape({9, 3, 1, 4}); in TEST() local
212 RuntimeShape input_shape({1, 3, 1, 4}); in TEST() local
[all …]
Dresize_nearest_neighbor_test.cc30 const RuntimeShape& input_shape, const std::vector<T>& input_data, in TestReferenceResizeNearestNeighbor()
49 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
60 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
72 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
84 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
95 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
106 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
118 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST() local
130 RuntimeShape input_shape = {1, 3, 3, 1}; in TEST() local
141 RuntimeShape input_shape = {1, 3, 3, 1}; in TEST() local
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_test.cc38 const std::vector<int32_t> input_shape{ variable
59 const std::vector<int32_t> input_shape{ variable
80 const std::vector<int32_t> input_shape{ variable
101 const std::vector<int32_t> input_shape{ variable
122 const std::vector<int32_t> input_shape{{shape_rng(), shape_rng()}}; variable
142 const std::vector<int32_t> input_shape{{shape_rng(), shape_rng()}}; variable
212 const std::vector<int32_t> input_shape{ in TEST() local
Dtranspose_tester.h44 inline TransposeTester& input_shape(std::vector<int32_t> input_shape) { in input_shape() function
49 inline const std::vector<int32_t>& input_shape() const { in input_shape() function
/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dreduction_benchmark.cc42 std::string GetReductionIR(StringRef op_name, ArrayRef<int32_t> input_shape, in GetReductionIR()
62 std::string GetSumF32IR(llvm::ArrayRef<int32_t> input_shape, in GetSumF32IR()
69 std::string GetMeanF32IR(llvm::ArrayRef<int32_t> input_shape, in GetMeanF32IR()
77 llvm::ArrayRef<ssize_t> input_shape) { in GetInputSpec()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_test.cc77 const RuntimeShape input_shape = ::tflite::GetTensorShape(shape); in ComputeExpectedTransposeResult() local
97 std::vector<int> input_shape = {3}; in TEST() local
110 std::vector<int> input_shape = {3}; in TEST() local
122 std::vector<int> input_shape = {3, 2}; in TEST() local
135 std::vector<int> input_shape = {3, 2}; in TEST() local
148 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
161 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
/external/ComputeLibrary/tests/validation/fixtures/
DWinogradConvolutionLayerFixture.h59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
166 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup()
297 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
338 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
359 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type) in setup()
411 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
453 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
522 …TensorType compute_target(const TensorShape &input_shape, const WinogradInfo &winograd_info, DataT… in compute_target()
[all …]
DFillFixture.h46 void setup(TensorShape input_shape, DataType data_type) in setup()
53 TensorType compute_target(const TensorShape &input_shape, DataType data_type) in compute_target()
75 SimpleTensor<T> compute_reference(const TensorShape &input_shape, DataType data_type) in compute_reference()
DGatherFixture.h51 void setup(TensorShape input_shape, TensorShape indices_shape, int axis, DataType data_type) in setup()
65 …void generate_indices(U &&indices, const TensorShape &input_shape, uint32_t actual_axis, TensorSha… in generate_indices()
78 TensorType compute_target(const TensorShape &input_shape, in compute_target()
117 SimpleTensor<T> compute_reference(const TensorShape &input_shape, in compute_reference()
DDirectConvolutionLayerFixture.h77 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
149 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
206 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
234 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
247 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
260 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
273 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DCopyFixture.h47 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 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
DPermuteFixture.h49 void setup(TensorShape input_shape, PermutationVector perm, DataType data_type) in setup()
62 …TensorType compute_target(const TensorShape &input_shape, DataType data_type, PermutationVector pe… in compute_target()
95 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, DataType data_type, PermutationV… in compute_reference()
DReshapeLayerFixture.h48 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 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
DSpaceToBatchFixture.h43 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
63 …TensorType compute_target(TensorShape input_shape, const TensorShape &block_shape_shape, const Ten… in compute_target()
115 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &block_shape_s… in compute_reference()
144 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
156 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
DROIAlignLayerFixture.h49 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
115 TensorType compute_target(TensorShape input_shape, in compute_target()
164 SimpleTensor<T> compute_reference(const TensorShape &input_shape, in compute_reference()
193 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
205 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
DROIPoolingLayerFixture.h49 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
104 TensorType compute_target(TensorShape input_shape, in compute_target()
149 SimpleTensor<T> compute_reference(const TensorShape &input_shape, in compute_reference()
178 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
191 …void setup(TensorShape input_shape, const ROIPoolingLayerInfo pool_info, TensorShape rois_shape, D… in setup()
DReorgLayerFixture.h47 void setup(TensorShape input_shape, int32_t stride, DataType data_type, DataLayout data_layout) in setup()
60 …TensorType compute_target(TensorShape input_shape, int32_t stride, DataType data_type, DataLayout … in compute_target()
96 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, int32_t stride, DataType data_ty… in compute_reference()
/external/tensorflow/tensorflow/lite/kernels/
Dreshape_test_common.h47 ReshapeOpModel(std::initializer_list<int> input_shape, in ReshapeOpModel()
81 void BuildWithHardcodedShape(std::initializer_list<int> input_shape, in BuildWithHardcodedShape()
94 void BuildWithTensorShape(std::initializer_list<int> input_shape, in BuildWithTensorShape()
114 void BuildWithConstantTensorShape(std::initializer_list<int> input_shape, in BuildWithConstantTensorShape()
/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
DPool2dFixture.h55 …void setup(TensorShape input_shape, const Pool2dAttributes &pool_attr, DataType data_type, bool mi… in setup()
85 …TensorType compute_target(TensorShape input_shape, const Pool2dAttributes &pool_attr, const DataTy… in compute_target()
155 …void setup(TensorShape input_shape, PoolingType pool_type, Size2D pool_size, Padding2D pad, Size2D… in setup()
168 …void setup(TensorShape input_shape, PoolingType pool_type, Size2D pool_size, Padding2D pad, Size2D… in setup()
181 void setup(TensorShape input_shape, Pool2dAttributes pool_attr, DataType data_type) in setup()
DDirectConv2dFixture.h92 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
169 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
207 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape output_shape, TensorSha… in setup()
226 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
248 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
320 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
345 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops_test.cc71 const TensorShape input_shape({4, 4}); in TEST() local
98 const TensorShape input_shape({4, 4}); in TEST() local
132 Status CreateSplitTensorGraph(const TensorShape& input_shape, in CreateSplitTensorGraph()
159 Status CreateSplitResourceGraph(const TensorShape& input_shape, in CreateSplitResourceGraph()
214 const TensorShape input_shape({1, 1, 1}); in TEST_P() local
232 const TensorShape input_shape({1, 1, 1}); in TEST_P() local
250 const TensorShape input_shape({2}); in TEST_P() local
268 const TensorShape input_shape({2, 2}); in TEST_P() local
285 const TensorShape input_shape({2, 2}); in TEST_P() local
303 const TensorShape input_shape({}); in TEST_P() local
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dadvanced_activations.py87 def compute_output_shape(self, input_shape): argument
145 def build(self, input_shape): argument
181 def compute_output_shape(self, input_shape): argument
225 def compute_output_shape(self, input_shape): argument
273 def compute_output_shape(self, input_shape): argument
358 def compute_output_shape(self, input_shape): argument
448 def compute_output_shape(self, input_shape): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc45 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
77 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
181 const TensorShape input_shape = ctx->InputShape(i); in Compile() local
218 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
234 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
261 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
311 StatusOr<xla::Shape> input_shape = ctx->builder()->GetShape(ctx->Input(0)); in Compile() local
416 auto input_shape = ctx->InputXlaShape(0).ValueOrDie(); in Compile() local
439 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dnth_element_op_test.py60 def _testFloatLargeInput(self, input_shape): argument
71 def _testIntLargeInput(self, input_shape): argument
82 def _testLargeInput(self, input_shape): argument
95 def _testEnumerateN(self, input_shape): argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpad.h39 const RuntimeShape& input_shape, const T* input_data, in PadImpl()
119 const RuntimeShape& input_shape, const T* input_data, in Pad()
129 const RuntimeShape& input_shape, const T* input_data, in Pad()
140 const RuntimeShape& input_shape, const int32_t* input_data, in Pad()
149 const RuntimeShape& input_shape, const T* input_data, in PadImageStyle()
158 const RuntimeShape& input_shape, in PadImageStyle()

12345678910>>...24