| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | whitespace_tokenizer_test.cc | 46 : input_shape_(input_shape) { in WhitespaceTokenizerModel()
50 for (int i = 0; i < input_shape_.size(); ++i) { in WhitespaceTokenizerModel()
78 for (int i = 0; i < input_shape_.size(); ++i) { in CheckRowSplits()
79 size *= input_shape_[i]; in CheckRowSplits()
84 if (i == input_shape_.size() - 1) { in CheckRowSplits()
95 expected_values[j] = j * input_shape_[i + 1]; in CheckRowSplits()
106 std::vector<int> input_shape_; member in tflite::ops::custom::whitespace_tokenizer::test::WhitespaceTokenizerModel
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | arg_min_max_test.cc | 46 : ArgBaseOpModel(input_type /*input_type*/), input_shape_(input_shape) {} in ArgMinOpModel()
52 BuildInterpreter({input_shape_, {1}}); in Build()
56 std::vector<int> input_shape_; member in tflite::ArgMinOpModel
62 : ArgBaseOpModel(input_type /*input_type*/), input_shape_(input_shape) {} in ArgMaxOpModel()
68 BuildInterpreter({input_shape_, {1}}); in Build()
72 std::vector<int> input_shape_; member in tflite::ArgMaxOpModel
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 27 input_shape_ = std::vector<size_t>(input_shape); in input_shape()
32 return input_shape_; in input_shape()
36 return i < input_shape_.size() ? input_shape_[i] : 1; in input_dim()
40 return input_shape_.size(); in num_dims()
45 input_shape_.cbegin(), input_shape_.cend(), size_t(1), std::multiplies<size_t>()); in num_input_elements()
221 std::vector<size_t> input_shape_;
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | fully_connected_tester.h | 40 input_shape_ = std::vector<int32_t>(shape.begin(), shape.end()); in InputShape()
41 input_size_ = ComputeSize(input_shape_); in InputShape()
45 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape()
107 std::vector<int32_t> input_shape_; variable
|
| D | reduce_tester.h | 40 input_shape_ = std::vector<int32_t>(shape.begin(), shape.end()); in InputShape()
41 input_size_ = ReduceTester::ComputeSize(input_shape_); in InputShape()
45 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape()
107 std::vector<int32_t> input_shape_; variable
|
| D | prelu_tester.h | 38 input_shape_ = std::vector<int32_t>(shape.begin(), shape.end()); in InputShape()
42 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape()
79 std::vector<int32_t> input_shape_; variable
|
| D | reshape_tester.h | 38 input_shape_ = std::vector<int32_t>(input_shape.begin(), input_shape.end()); in InputShape()
43 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape()
77 std::vector<int32_t> input_shape_; variable
|
| D | pad_tester.h | 38 input_shape_ = std::vector<int32_t>(shape.begin(), shape.end()); in InputShape()
42 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape()
79 std::vector<int32_t> input_shape_; variable
|
| D | fully_connected_tester.cc | 39 EXPECT_NE(input_shape_.size(), 0); in OutputShape()
41 std::vector<int32_t> output_shape(input_shape_.cbegin(), in OutputShape()
42 input_shape_.cend() - 1); in OutputShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 114 input_shape_(input_shape), in FftThunk()
122 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape_); in ExecuteOnStream()
149 for (int i = 0; i < input_shape_.dimensions_size() - fft_rank; ++i) { in ExecuteOnStream()
150 batch_size *= input_shape_.dimensions(i); in ExecuteOnStream()
161 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; in ExecuteOnStream()
163 input_embed[i] = input_shape_.dimensions(dim_offset); in ExecuteOnStream()
164 input_distance *= input_shape_.dimensions(dim_offset); in ExecuteOnStream()
|
| D | fft_thunk.h | 96 const Shape input_shape_; variable
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite_test.cc | 50 input_shape_ = AddInput(index_type); in RaggedTensorToTensorOpModel()
87 PopulateTensor(input_shape_, shape); in InvokeFloat()
98 PopulateTensor(input_shape_, shape); in InvokeInt()
108 int input_shape_; member in tflite::__anon79817ba20111::RaggedTensorToTensorOpModel
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_multithread.h | 43 input_shape_(input_shape), in DepthwiseConvWorkerTask()
57 DepthwiseConvImpl(params_, input_shape_, input_data_, filter_shape_, in Run()
65 const RuntimeShape& input_shape_; member
|
| D | legacy_optimized_ops.h | 389 input_shape_(input_shape), in LegacyDepthwiseConvWorkerTask()
402 LegacyDepthwiseConvImpl(params_, input_shape_, input_data_, filter_shape_, in Run()
410 const RuntimeShape& input_shape_; member
501 input_shape_(input_shape), in LegacyPerChannelDepthwiseConvWorkerTask()
516 params_, output_multiplier_, output_shift_, input_shape_, input_data_, in Run()
525 const RuntimeShape& input_shape_; member
998 : input_shape_(input_shape), in LegacyFullyConnectedAsGEMVWorkerTask()
1018 input_shape_, input_data_, input_offset_, filter_shape_, filter_data_, in Run()
1025 const RuntimeShape& input_shape_; member
2009 : input_shape_(input_shape), in LegacyInt8FullyConnectedAsGEMVWorkerTask()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | winograd_transform.h | 34 : filter_shape_(3, 3), input_shape_(4, 4), output_shape_(2, 2) {} in WinogradTransform()
46 virtual const Shape& input_shape() const { return input_shape_; } in input_shape()
51 const Shape input_shape_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | depthwise_conv.h | 1907 input_shape_(input_shape),
1921 DepthwiseConvImpl(params_, output_multiplier_, output_shift_, input_shape_,
1931 const RuntimeShape& input_shape_;
|