| /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 | 30 def _get_const_output_shape(input_shape, dim): argument
166 def test_conv_kernel_mask_fc(self, *input_shape):
183 def test_conv_kernel_mask_diag(self, *input_shape):
201 def test_conv_kernel_mask_full_stride(self, *input_shape):
222 def test_conv_kernel_mask_almost_full_stride(self, *input_shape):
245 def test_conv_kernel_mask_rect_kernel(self, *input_shape):
278 def test_conv_kernel_mask_wrong_padding(self, *input_shape):
301 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/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/lite/micro/kernels/ |
| D | strided_slice_test.cc | 59 void TestStridedSliceFloat(const int* input_shape, const int* begin_shape, in TestStridedSliceFloat()
91 const int* input_shape, const int* begin_shape, const int* end_shape, in TestStridedSliceQuantized()
127 const int input_shape[] = {5, 2, 2, 2, 2, 2}; in TF_LITE_MICRO_TEST() local
148 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
169 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
190 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
211 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
232 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
253 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
274 const int input_shape[] = {1, 4}; in TF_LITE_MICRO_TEST() local
[all …]
|
| D | shape_test.cc | 73 int input_shape[] = {1, 5}; in TF_LITE_MICRO_TEST() local
84 int input_shape[] = {2, 4, 3}; in TF_LITE_MICRO_TEST() local
95 int input_shape[] = {2, 12, 1}; in TF_LITE_MICRO_TEST() local
106 int input_shape[] = {2, 2, 6}; in TF_LITE_MICRO_TEST() local
117 int input_shape[] = {2, 2, 2, 3}; in TF_LITE_MICRO_TEST() local
128 int input_shape[] = {1, 1}; in TF_LITE_MICRO_TEST() local
|
| D | pooling_test.cc | 185 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
201 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
223 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
245 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
268 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
290 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
312 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
334 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
350 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
366 const int input_shape[] = {4, 1, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
[all …]
|
| /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
|
| D | dynamic_dimension_inference_test.cc | 105 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
128 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
151 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
180 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
203 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
235 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
268 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F() local
488 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F() local
527 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F() local
566 auto input_shape = ShapeUtil::MakeShape(F32, {2, 3, 4, 5, 6}); in TEST_F() local
[all …]
|
| /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/python/keras/applications/ |
| D | efficientnet.py | 207 input_shape=None, argument
526 input_shape=None, argument
552 input_shape=None, argument
578 input_shape=None, argument
604 input_shape=None, argument
630 input_shape=None, argument
656 input_shape=None, argument
682 input_shape=None, argument
708 input_shape=None, argument
|
| D | densenet.py | 134 input_shape=None, argument
326 input_shape=None, argument
339 input_shape=None, argument
352 input_shape=None, argument
|
| D | resnet_v2.py | 37 input_shape=None, argument
68 input_shape=None, argument
99 input_shape=None, 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
153 const TensorShape input_shape = ctx->InputShape(i); in Compile() local
190 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
206 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
233 const TensorShape input_shape = ctx->InputShape("input"); in Compile() local
283 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
382 auto input_shape = ctx->InputXlaShape(0).ValueOrDie(); in Compile() local
405 const TensorShape input_shape = ctx->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | advanced_activations.py | 89 def compute_output_shape(self, input_shape): argument
147 def build(self, input_shape): argument
183 def compute_output_shape(self, input_shape): argument
226 def compute_output_shape(self, input_shape): argument
268 def compute_output_shape(self, input_shape): argument
353 def compute_output_shape(self, input_shape): argument
442 def compute_output_shape(self, input_shape): argument
|
| D | merge.py | 90 def build(self, input_shape): argument
186 def compute_output_shape(self, input_shape): argument
283 def build(self, input_shape): argument
490 def build(self, input_shape): argument
525 def compute_output_shape(self, input_shape): argument
660 def build(self, input_shape): argument
707 def compute_output_shape(self, input_shape): argument
|
| D | local.py | 159 def build(self, input_shape): argument
250 def compute_output_shape(self, input_shape): argument
461 def build(self, input_shape): argument
555 def compute_output_shape(self, input_shape): argument
635 def get_locallyconnected_mask(input_shape, kernel_shape, strides, padding, argument
|
| D | einsum_dense.py | 140 def build(self, input_shape): argument
209 def _analyze_einsum_string(equation, bias_axes, input_shape, output_shape): argument
242 input_shape, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | pad.h | 39 const RuntimeShape& input_shape, const T* input_data, in PadImpl()
112 const RuntimeShape& input_shape, const T* input_data, in Pad()
122 const RuntimeShape& input_shape, const T* input_data, in Pad()
133 const RuntimeShape& input_shape, const int32_t* input_data, in Pad()
142 const RuntimeShape& input_shape, const T* input_data, in PadImageStyle()
151 const RuntimeShape& input_shape, in PadImageStyle()
|
| D | logistic.h | 30 inline void Logistic(const RuntimeShape& input_shape, const float* input_data, in Logistic()
61 inline void Logistic(const LogisticParams&, const RuntimeShape& input_shape, in Logistic()
69 const RuntimeShape& input_shape, const int16_t* input_data, in Logistic()
93 inline void Logistic(const RuntimeShape& input_shape, const int8_t* input_data, in Logistic()
|
| D | tanh.h | 29 inline void Tanh(const RuntimeShape& input_shape, const float* input_data, in Tanh()
42 inline void Tanh(const TanhParams&, const RuntimeShape& input_shape, in Tanh()
49 inline void Tanh(const TanhParams& params, const RuntimeShape& input_shape, in Tanh()
83 inline void Tanh(const TanhParams& params, const RuntimeShape& input_shape, in Tanh()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | nth_element_op_test.py | 64 def _testFloatLargeInput(self, input_shape): argument
75 def _testIntLargeInput(self, input_shape): argument
86 def _testLargeInput(self, input_shape): argument
99 def _testEnumerateN(self, input_shape): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 84 auto input_shape = c->input(0); in __anon815c5c9f0302() local
127 auto input_shape = c->input(0); in __anon815c5c9f0402() local
174 auto input_shape = c->input(0); in __anon815c5c9f0502() local
225 auto input_shape = c->input(0); in __anon815c5c9f0602() local
262 auto input_shape = c->input(0); in __anon815c5c9f0702() local
302 auto input_shape = c->input(0); in __anon815c5c9f0802() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_integration_test.py | 128 def test_layer_output_type(self, layer_to_test, input_shape, _, layer_kwargs): argument
149 def _run_fit_eval_predict(self, layer_to_test, input_shape, data_shape, argument
189 def test_model_loops(self, layer_to_test, input_shape, fuzz_dims, argument
|