Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darg_min_max_test.cc56 std::vector<int> input_shape_; member in tflite::ArgMinOpModel
72 std::vector<int> input_shape_; member in tflite::ArgMaxOpModel
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dpad_tester.h79 std::vector<int32_t> input_shape_; variable
Dprelu_tester.h79 std::vector<int32_t> input_shape_; variable
Dreshape_tester.h77 std::vector<int32_t> input_shape_; variable
Dfully_connected_tester.h107 std::vector<int32_t> input_shape_; variable
Dreduce_tester.h107 std::vector<int32_t> input_shape_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.h96 const Shape input_shape_; variable
/external/tensorflow/tensorflow/core/kernels/
Dwinograd_transform.h51 const Shape input_shape_; variable
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dwhitespace_tokenizer_test.cc106 std::vector<int> input_shape_; member in tflite::ops::custom::whitespace_tokenizer::test::WhitespaceTokenizerModel
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_multithread.h65 const RuntimeShape& input_shape_; member
Dlegacy_optimized_ops.h410 const RuntimeShape& input_shape_; member
525 const RuntimeShape& input_shape_; member
1025 const RuntimeShape& input_shape_; member
2036 const RuntimeShape& input_shape_; member
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite_test.cc108 int input_shape_; member in tflite::__anon79817ba20111::RaggedTensorToTensorOpModel