/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
H A D | arg_min_max_test.cc | 58 std::vector<int> input_shape_; member in tflite::ArgMinOpModel 74 std::vector<int> input_shape_; member in tflite::ArgMaxOpModel
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
H A D | transpose_tester.h | 70 std::vector<int32_t> input_shape_; variable
|
H A D | pad_tester.h | 79 std::vector<int32_t> input_shape_; variable
|
H A D | split_tester.h | 80 std::vector<int32_t> input_shape_; variable
|
H A D | reshape_tester.h | 77 std::vector<int32_t> input_shape_; variable
|
H A D | reduce_tester.h | 107 std::vector<int32_t> input_shape_; variable
|
H A D | prelu_tester.h | 102 std::vector<int32_t> input_shape_; variable
|
H A D | quantized_pad_tester.h | 108 std::vector<int32_t> input_shape_; variable
|
H A D | fully_connected_tester.h | 142 std::vector<int32_t> input_shape_; variable
|
H A D | quantized_reduce_tester.h | 141 std::vector<int32_t> input_shape_; variable
|
H A D | quantized_fully_connected_tester.h | 173 std::vector<int32_t> input_shape_; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | fft_thunk.h | 87 const Shape input_shape_; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | winograd_transform.h | 51 const Shape input_shape_; variable
|
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
H A D | whitespace_tokenizer_test.cc | 106 std::vector<int> input_shape_; member in tflite::ops::custom::whitespace_tokenizer::test::WhitespaceTokenizerModel
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | depthwiseconv_multithread.h | 67 const RuntimeShape& input_shape_; member
|
H A D | legacy_optimized_ops.h | 411 const RuntimeShape& input_shape_; member 526 const RuntimeShape& input_shape_; member 1026 const RuntimeShape& input_shape_; member 2037 const RuntimeShape& input_shape_; member
|
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
H A D | ragged_tensor_to_tensor_tflite_test.cc | 108 int input_shape_; member in tflite::__anoneeaa06110111::RaggedTensorToTensorOpModel
|