Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darg_min_max_test.cc58 std::vector<int> input_shape_; member in tflite::ArgMinOpModel
74 std::vector<int> input_shape_; member in tflite::ArgMaxOpModel
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dtranspose_tester.h70 std::vector<int32_t> input_shape_; variable
Dpad_tester.h79 std::vector<int32_t> input_shape_; variable
Dsplit_tester.h80 std::vector<int32_t> input_shape_; variable
Dreshape_tester.h77 std::vector<int32_t> input_shape_; variable
Dprelu_tester.h102 std::vector<int32_t> input_shape_; variable
Dreduce_tester.h107 std::vector<int32_t> input_shape_; variable
Dquantized_pad_tester.h108 std::vector<int32_t> input_shape_; variable
Dfully_connected_tester.h142 std::vector<int32_t> input_shape_; variable
Dquantized_reduce_tester.h141 std::vector<int32_t> input_shape_; variable
Dquantized_fully_connected_tester.h173 std::vector<int32_t> input_shape_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.h87 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.h67 const RuntimeShape& input_shape_; member
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
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite_test.cc108 int input_shape_; member in tflite::__anoncaded0910111::RaggedTensorToTensorOpModel