Home
last modified time | relevance | path

Searched defs:input_shape_ (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Darg_min_max_test.cc58 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 Dtranspose_tester.h70 std::vector<int32_t> input_shape_; variable
H A Dpad_tester.h79 std::vector<int32_t> input_shape_; variable
H A Dsplit_tester.h80 std::vector<int32_t> input_shape_; variable
H A Dreshape_tester.h77 std::vector<int32_t> input_shape_; variable
H A Dreduce_tester.h107 std::vector<int32_t> input_shape_; variable
H A Dprelu_tester.h102 std::vector<int32_t> input_shape_; variable
H A Dquantized_pad_tester.h108 std::vector<int32_t> input_shape_; variable
H A Dfully_connected_tester.h142 std::vector<int32_t> input_shape_; variable
H A Dquantized_reduce_tester.h141 std::vector<int32_t> input_shape_; variable
H A Dquantized_fully_connected_tester.h173 std::vector<int32_t> input_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dfft_thunk.h87 const Shape input_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dwinograd_transform.h51 const Shape input_shape_; variable
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dwhitespace_tokenizer_test.cc106 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 Ddepthwiseconv_multithread.h67 const RuntimeShape& input_shape_; member
H A Dlegacy_optimized_ops.h411 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 Dragged_tensor_to_tensor_tflite_test.cc108 int input_shape_; member in tflite::__anoneeaa06110111::RaggedTensorToTensorOpModel