| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 24 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 …]
|
| D | resize_nearest_neighbor_test.cc | 30 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/python/keras/utils/ |
| D | conv_utils_test.py | 26 def _get_const_output_shape(input_shape, dim): argument
162 def test_conv_kernel_mask_fc(self, *input_shape):
179 def test_conv_kernel_mask_diag(self, *input_shape):
197 def test_conv_kernel_mask_full_stride(self, *input_shape):
218 def test_conv_kernel_mask_almost_full_stride(self, *input_shape):
241 def test_conv_kernel_mask_rect_kernel(self, *input_shape):
274 def test_conv_kernel_mask_wrong_padding(self, *input_shape):
297 def test_conv_kernel_mask_wrong_dims(self, *input_shape):
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_test.cc | 38 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
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | WinogradConvolutionLayerFixture.h | 59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
88 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
129 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
163 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
197 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_sha… in compute_target()
246 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
326 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup()
354 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
387 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
407 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type) in setup()
[all …]
|
| D | UpsampleLayerFixture.h | 47 void setup(TensorShape input_shape, DataType data_type, DataLayout data_layout, in setup()
63 …TensorType compute_target(TensorShape input_shape, const Size2D &info, const InterpolationPolicy &… in compute_target()
103 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const Size2D &info, const Interp… in compute_reference()
125 void setup(TensorShape input_shape, DataType data_type, DataLayout data_layout, in setup()
138 void setup(TensorShape input_shape, DataType data_type, DataLayout data_layout, in setup()
|
| D | FillFixture.h | 46 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()
|
| D | DirectConvolutionLayerFixture.h | 55 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
76 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
128 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &b… in compute_target()
175 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
202 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
215 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
228 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
241 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
|
| D | DirectConvolutionLayerTensorShiftFixture.h | 52 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
69 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
113 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
164 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
210 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
222 …void setup(TensorShape input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int… in setup()
234 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
247 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
|
| D | GatherFixture.h | 51 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()
|
| D | SpaceToBatchFixture.h | 43 …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()
|
| D | FullyConnectedLayerFixture.h | 58 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
103 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const … in compute_target()
193 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
221 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
235 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… 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 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… 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 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape,… in compute_reference()
|
| D | PermuteFixture.h | 49 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()
|
| D | ROIAlignLayerFixture.h | 49 …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()
|
| D | DeconvolutionLayerFixture.h | 53 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
126 …TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape bi… in compute_target()
185 …SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &weights_shape… in compute_reference()
223 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int padx, unsigned … in setup()
244 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int pad_left, unsig… in setup()
265 …void setup(TensorShape input_shape, unsigned int sx, unsigned int sy, unsigned int padx, unsigned … in setup()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_test.cc | 75 const RuntimeShape input_shape = ::tflite::GetTensorShape(shape); in ComputeExpectedTransposeResult() local
95 std::vector<int> input_shape = {3}; in TEST() local
108 std::vector<int> input_shape = {3}; in TEST() local
120 std::vector<int> input_shape = {3, 2}; in TEST() local
133 std::vector<int> input_shape = {3, 2}; in TEST() local
146 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
159 std::vector<int> input_shape = {2, 2, 3, 1}; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util_test.cc | 37 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 3, 5}); in TEST() local
47 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 3, 5}); in TEST() local
56 Shape input_shape = ShapeUtil::MakeShape(F32, {4, 5, 7}); in TEST() local
68 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7}); in TEST() local
80 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7}); in TEST() local
94 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 4, 7}); in TEST() local
107 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7, 5, 3}); in TEST() local
120 Shape input_shape = ShapeUtil::MakeShape(F32, {3, 1, 5, 7}); in TEST() local
133 Shape input_shape = ShapeUtil::MakeShape(F32, {8, 16}); in TEST() local
153 Shape input_shape = ShapeUtil::MakeShape(F32, {1, 1, 1}); in TEST() local
|
| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | efficientnet.py | 219 input_shape=None, argument
530 input_shape=None, argument
556 input_shape=None, argument
582 input_shape=None, argument
608 input_shape=None, argument
634 input_shape=None, argument
660 input_shape=None, argument
686 input_shape=None, argument
712 input_shape=None, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test_common.h | 47 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/tensorflow/tensorflow/core/tpu/kernels/ |
| D | sharding_util_ops_test.cc | 71 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/ |
| D | advanced_activations.py | 87 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/ |
| D | shape_op.cc | 45 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 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
410 auto input_shape = ctx->InputXlaShape(0).ValueOrDie(); in Compile() local
433 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | pad.h | 39 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()
|