| /external/tensorflow/tensorflow/lite/kernels/ |
| D | select_test.cc | 91 model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); in TEST()
92 model.PopulateTensor<bool>(model.input2(), {false, false, false, false}); in TEST()
93 model.PopulateTensor<bool>(model.input3(), {true, true, true, true}); in TEST()
105 model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); in TEST()
106 model.PopulateTensor<float>(model.input2(), {0.1, 0.2, 0.3, 0.4}); in TEST()
107 model.PopulateTensor<float>(model.input3(), {0.5, 0.6, 0.7, 0.8}); in TEST()
118 model.PopulateTensor<bool>(model.input1(), {false, true, false, false}); in TEST()
119 model.PopulateTensor<uint8_t>(model.input2(), {1, 2, 3, 4}); in TEST()
120 model.PopulateTensor<uint8_t>(model.input3(), {5, 6, 7, 8}); in TEST()
131 model.PopulateTensor<bool>(model.input1(), {false, true, false, false}); in TEST()
[all …]
|
| D | unsorted_segment_test.cc | 34 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST_P()
35 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1}); in TEST_P()
36 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST_P()
44 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4}); in TEST_P()
45 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1}); in TEST_P()
46 model.PopulateTensor<int32_t>(model.num_segments(), {1}); in TEST_P()
53 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST_P()
54 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST_P()
55 model.PopulateTensor<int32_t>(model.num_segments(), {2, 1}); in TEST_P()
62 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST_P()
[all …]
|
| D | mirror_pad_test.cc | 57 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
58 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {0, 0, 0, 0}); in TEST()
67 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
68 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {0, 1, 0, 1}); in TEST()
78 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
79 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {1, 0, 1, 0}); in TEST()
89 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
90 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {0, 1, 0, 1}); in TEST()
100 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
101 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {1, 0, 1, 0}); in TEST()
[all …]
|
| D | sparse_to_dense_test.cc | 66 PopulateTensor<int32_t>(output_shape_, output_shape_data); in SparseToDenseOpModel()
69 PopulateTensor<T>(default_value_, {default_value}); in SparseToDenseOpModel()
98 m.PopulateTensor<int32_t>(m.indices(), {3}); in TEST_P()
99 m.PopulateTensor<int32_t>(m.output_shape(), {5}); in TEST_P()
100 m.PopulateTensor<float>(m.values(), {7}); in TEST_P()
110 m.PopulateTensor<int32_t>(m.indices(), {1, 3, 5}); in TEST_P()
111 m.PopulateTensor<int32_t>(m.output_shape(), {7}); in TEST_P()
112 m.PopulateTensor<float>(m.values(), {2, 4, 6}); in TEST_P()
122 m.PopulateTensor<int32_t>(m.indices(), {0, 0, 0, 1, 2, 1, 2, 0, 1}); in TEST_P()
123 m.PopulateTensor<int32_t>(m.output_shape(), {3, 3, 3}); in TEST_P()
[all …]
|
| D | unsorted_segment_sum_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {5, 1, 7, 2, 3, 4, 10}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1, 1, 0, 1, 0}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), in TEST()
72 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
73 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
83 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 4.0, 3.0}); in TEST()
84 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7}); in TEST()
85 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
97 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, in TEST()
[all …]
|
| D | range_test.cc | 57 model.PopulateTensor<int32_t>(model.start(), {0}); in TEST()
58 model.PopulateTensor<int32_t>(model.limit(), {4}); in TEST()
59 model.PopulateTensor<int32_t>(model.delta(), {1}); in TEST()
67 model.PopulateTensor<int32_t>(model.start(), {2}); in TEST()
68 model.PopulateTensor<int32_t>(model.limit(), {9}); in TEST()
69 model.PopulateTensor<int32_t>(model.delta(), {2}); in TEST()
77 model.PopulateTensor<int32_t>(model.start(), {10}); in TEST()
78 model.PopulateTensor<int32_t>(model.limit(), {3}); in TEST()
79 model.PopulateTensor<int32_t>(model.delta(), {-3}); in TEST()
87 model.PopulateTensor<float>(model.start(), {0}); in TEST()
[all …]
|
| D | reverse_test.cc | 59 model.PopulateTensor<float>(model.input(), {1, 2, 3, 4}); in TEST()
60 model.PopulateTensor<int32_t>(model.axis(), {0}); in TEST()
70 model.PopulateTensor<float>(model.input(), in TEST()
73 model.PopulateTensor<int32_t>(model.axis(), {1}); in TEST()
87 model.PopulateTensor<int32_t>(model.input(), {1, 2, 3, 4}); in TEST()
88 model.PopulateTensor<int32_t>(model.axis(), {0}); in TEST()
98 model.PopulateTensor<int32_t>( in TEST()
101 model.PopulateTensor<int32_t>(model.axis(), {1}); in TEST()
115 model.PopulateTensor<int64_t>(model.input(), {1, 2, 3, 4}); in TEST()
116 model.PopulateTensor<int32_t>(model.axis(), {0}); in TEST()
[all …]
|
| D | segment_sum_test.cc | 53 model.PopulateTensor<int32_t>(model.data(), in TEST()
55 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1}); in TEST()
64 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3}); in TEST()
65 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1}); in TEST()
74 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST()
75 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1}); in TEST()
84 model.PopulateTensor<float>(model.data(), in TEST()
86 model.PopulateTensor<int>(model.segment_ids(), {0, 0, 1}); in TEST()
96 model.PopulateTensor<float>(model.data(), {1, 2, 3}); in TEST()
97 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1}); in TEST()
[all …]
|
| D | unidirectional_sequence_lstm_test_util.h | 135 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights()
139 PopulateTensor(input_to_forget_weights_, f); in SetInputToForgetWeights()
143 PopulateTensor(input_to_cell_weights_, f); in SetInputToCellWeights()
147 PopulateTensor(input_to_output_weights_, f); in SetInputToOutputWeights()
151 PopulateTensor(recurrent_to_input_weights_, f); in SetRecurrentToInputWeights()
155 PopulateTensor(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
159 PopulateTensor(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
163 PopulateTensor(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
167 PopulateTensor(cell_to_input_weights_, f); in SetCellToInputWeights()
171 PopulateTensor(cell_to_forget_weights_, f); in SetCellToForgetWeights()
[all …]
|
| D | reverse_sequence_test.cc | 61 model.PopulateTensor<float>(model.input(), in TEST()
64 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 2, 3, 3}); in TEST()
76 model.PopulateTensor<float>(model.input(), in TEST()
79 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 4}); in TEST()
91 model.PopulateTensor<int32_t>( in TEST()
94 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 2, 3, 3}); in TEST()
106 model.PopulateTensor<int32_t>( in TEST()
109 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 4}); in TEST()
121 model.PopulateTensor<int64_t>( in TEST()
124 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 2, 3, 3}); in TEST()
[all …]
|
| D | unsorted_segment_prod_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 4, 3, 2, 1}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, 7}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 4, 3, 2, 1}); in TEST()
71 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, -1}); in TEST()
72 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
82 model.PopulateTensor<int32_t>(model.data(), in TEST()
84 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
85 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
95 model.PopulateTensor<float>(model.data(), in TEST()
[all …]
|
| D | comparisons_test.cc | 105 model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); in TEST()
106 model.PopulateTensor<bool>(model.input2(), {true, true, false, false}); in TEST()
116 model.PopulateTensor<float>(model.input1(), {0.1, 0.9, 0.7, 0.3}); in TEST()
117 model.PopulateTensor<float>(model.input2(), {0.1, 0.2, 0.6, 0.5}); in TEST()
127 model.PopulateTensor<int>(model.input1(), {-1, 9, 7, 3}); in TEST()
128 model.PopulateTensor<int>(model.input2(), {1, 2, 7, 5}); in TEST()
141 model.PopulateTensor<std::string>(model.input1(), {"A", "B", "C", "D"}); in TEST()
142 model.PopulateTensor<std::string>(model.input2(), {"A", "C", "B", "D"}); in TEST()
152 model.PopulateTensor<int>(model.input1(), {-1, 9, 7, 3}); in TEST()
153 model.PopulateTensor<int>(model.input2(), {7}); in TEST()
[all …]
|
| D | unsorted_segment_min_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {5, 3, 7, 8, 6, 4}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1, 1, 0, 1}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), in TEST()
72 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
73 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
83 model.PopulateTensor<float>(model.data(), in TEST()
85 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, 7}); in TEST()
86 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
102 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, in TEST()
[all …]
|
| D | unsorted_segment_max_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {5, 1, 7, 2, 3, 4}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1, 1, 0, 1}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), in TEST()
72 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
73 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
83 model.PopulateTensor<float>(model.data(), in TEST()
85 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, 7}); in TEST()
86 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
102 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, in TEST()
[all …]
|
| D | floor_mod_test.cc | 34 model.PopulateTensor<int32_t>(model.input1(), {10, 9, 11, 3}); in TEST()
35 model.PopulateTensor<int32_t>(model.input2(), {2, 2, 3, 4}); in TEST()
45 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
46 model.PopulateTensor<int32_t>(model.input2(), {2, 2, -3, -4}); in TEST()
55 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
56 model.PopulateTensor<int32_t>(model.input2(), {-3}); in TEST()
65 model.PopulateTensor<int64_t>(model.input1(), {10, -9, -11, (1LL << 34) + 9}); in TEST()
66 model.PopulateTensor<int64_t>(model.input2(), {-(1LL << 33)}); in TEST()
77 model.PopulateTensor<float>(model.input1(), {10, 9, 11, 3}); in TEST()
78 model.PopulateTensor<float>(model.input2(), {2, 2, 3, 4}); in TEST()
[all …]
|
| D | optional_tensor_test.cc | 112 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights()
116 PopulateTensor(input_to_forget_weights_, f); in SetInputToForgetWeights()
120 PopulateTensor(input_to_cell_weights_, f); in SetInputToCellWeights()
124 PopulateTensor(input_to_output_weights_, f); in SetInputToOutputWeights()
128 PopulateTensor(recurrent_to_input_weights_, f); in SetRecurrentToInputWeights()
132 PopulateTensor(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
136 PopulateTensor(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
140 PopulateTensor(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
144 PopulateTensor(cell_to_input_weights_, f); in SetCellToInputWeights()
148 PopulateTensor(cell_to_forget_weights_, f); in SetCellToForgetWeights()
[all …]
|
| D | cumsum_test.cc | 62 m.PopulateTensor<int>(m.input(), {1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
63 m.PopulateTensor<int>(m.axis(), {1}); in TEST()
75 m.PopulateTensor<int64_t>( in TEST()
78 m.PopulateTensor<int>(m.axis(), {1}); in TEST()
92 m.PopulateTensor<int>(m.input(), {1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
93 m.PopulateTensor<int>(m.axis(), {0}); in TEST()
105 m.PopulateTensor<int>(m.input(), {1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
106 m.PopulateTensor<int>(m.axis(), {0}); in TEST()
118 m.PopulateTensor<int>(m.input(), {1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
119 m.PopulateTensor<int>(m.axis(), {1}); in TEST()
[all …]
|
| D | cast_test.cc | 51 m.PopulateTensor<int16_t>(m.input(), {100, 200, 300, 400, 500, 600}); in TEST()
59 m.PopulateTensor<int16_t>(m.input(), {100, 200, 300, 400, 500, 600}); in TEST()
67 m.PopulateTensor<int32_t>(m.input(), {100, 200, 300, 400, 500, 600}); in TEST()
75 m.PopulateTensor<float>(m.input(), {100.f, 20.f, 3.f, 0.4f, 0.999f, 1.1f}); in TEST()
83 m.PopulateTensor<float>(m.input(), {100.f, 20.f, 3.f, 0.4f, 0.999f, 1.1f}); in TEST()
91 m.PopulateTensor<int64_t>(m.input(), {100, 200, 300, 400, 500, 600}); in TEST()
99 m.PopulateTensor<float>(m.input(), {100.f, 20.f, 3.f, 0.4f, 0.999f, 1.1f}); in TEST()
107 m.PopulateTensor<float>(m.input(), {100.f, -1.0f, 0.f, 0.4f, 0.999f, 1.1f}); in TEST()
115 m.PopulateTensor<bool>(m.input(), {true, true, false, true, false, true}); in TEST()
123 m.PopulateTensor<float>(m.input(), {100.f, 1.0f, 0.f, 0.4f, 1.999f, 1.1f}); in TEST()
[all …]
|
| D | floor_div_test.cc | 58 model.PopulateTensor<int32_t>(model.input1(), {10, 9, 11, 3}); in TEST()
59 model.PopulateTensor<int32_t>(model.input2(), {2, 2, 3, 4}); in TEST()
69 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
70 model.PopulateTensor<int32_t>(model.input2(), {2, 2, -3, -4}); in TEST()
79 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
80 model.PopulateTensor<int32_t>(model.input2(), {-3}); in TEST()
90 model.PopulateTensor<float>(model.input1(), {10.05, 9.09, 11.9, 3.01}); in TEST()
91 model.PopulateTensor<float>(model.input2(), {2.05, 2.03, 3.03, 4.03}); in TEST()
101 model.PopulateTensor<float>(model.input1(), {10.03, -9.9, -11.0, 7.0}); in TEST()
102 model.PopulateTensor<float>(model.input2(), {2.0, 2.3, -3.0, -4.1}); in TEST()
[all …]
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment_test.cc | 34 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST_P()
35 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1}); in TEST_P()
36 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST_P()
44 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4}); in TEST_P()
45 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1}); in TEST_P()
46 model.PopulateTensor<int32_t>(model.num_segments(), {1}); in TEST_P()
53 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST_P()
54 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST_P()
55 model.PopulateTensor<int32_t>(model.num_segments(), {2, 1}); in TEST_P()
62 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 5, 6}); in TEST_P()
[all …]
|
| D | unsorted_segment_sum_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {5, 1, 7, 2, 3, 4, 10}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1, 1, 0, 1, 0}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), in TEST()
72 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
73 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
83 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 4.0, 3.0}); in TEST()
84 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7}); in TEST()
85 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
97 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, in TEST()
[all …]
|
| D | unsorted_segment_prod_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 4, 3, 2, 1}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, 7}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), {1, 2, 3, 4, 4, 3, 2, 1}); in TEST()
71 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, -1}); in TEST()
72 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
82 model.PopulateTensor<int32_t>(model.data(), in TEST()
84 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
85 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
95 model.PopulateTensor<float>(model.data(), in TEST()
[all …]
|
| D | unsorted_segment_max_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {5, 1, 7, 2, 3, 4}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1, 1, 0, 1}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), in TEST()
72 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
73 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
83 model.PopulateTensor<float>(model.data(), in TEST()
85 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, 7}); in TEST()
86 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
102 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, in TEST()
[all …]
|
| D | unsorted_segment_min_test.cc | 58 model.PopulateTensor<int32_t>(model.data(), {5, 3, 7, 8, 6, 4}); in TEST()
59 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 0, 1, 1, 0, 1}); in TEST()
60 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
70 model.PopulateTensor<int32_t>(model.data(), in TEST()
72 model.PopulateTensor<int32_t>(model.segment_ids(), {0, 1, 0}); in TEST()
73 model.PopulateTensor<int32_t>(model.num_segments(), {2}); in TEST()
83 model.PopulateTensor<float>(model.data(), in TEST()
85 model.PopulateTensor<int32_t>(model.segment_ids(), {1, 0, 1, 7, 7, 7, 7, 7}); in TEST()
86 model.PopulateTensor<int32_t>(model.num_segments(), {8}); in TEST()
102 model.PopulateTensor<float>(model.data(), {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | elementwise_test.cc | 48 ASSERT_TRUE(model.PopulateTensor(0, {0.0, -6.2, 2.0, 4.0})); in TEST()
60 ASSERT_TRUE(model.PopulateTensor(0, {0.0, 3.1415926, -3.1415926, 1})); in TEST()
72 ASSERT_TRUE(model.PopulateTensor(0, {0.0, -6.2, 2.0, 4.0})); in TEST()
83 ASSERT_TRUE(model.PopulateTensor( in TEST()
98 ASSERT_TRUE(model.PopulateTensor( in TEST()
115 model.PopulateTensor(0, {-4.5f, -3.0f, -1.5f, 0.0f, 1.5f, 3.0f, 4.5f})); in TEST()
129 model.PopulateTensor(0, {-4.5f, -3.0f, -1.5f, 0.0f, 1.5f, 3.0f, 4.5f})); in TEST()
142 ASSERT_TRUE(model.PopulateTensor(0, {1.0, 3.1415926, 1.0, 1.0})); in TEST()
154 ASSERT_TRUE(model.PopulateTensor(0, {1.0, -3.1415926, 0.0, 1.0})); in TEST()
166 ASSERT_TRUE(model.PopulateTensor(0, {1.0, 2.0, 4.0, 9.0})); in TEST()
[all …]
|