Home
last modified time | relevance | path

Searched refs:PopulateTensor (Results 1 – 25 of 88) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad_test.cc56 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
57 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {0, 0, 0, 0}); in TEST()
66 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
67 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {0, 1, 0, 1}); in TEST()
77 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
78 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {1, 0, 1, 0}); in TEST()
88 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
89 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {0, 1, 0, 1}); in TEST()
99 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
100 model.PopulateTensor<int>(model.padding_matrix_tensor_id(), {1, 0, 1, 0}); in TEST()
[all …]
Dselect_test.cc63 model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); in TEST()
64 model.PopulateTensor<bool>(model.input2(), {false, false, false, false}); in TEST()
65 model.PopulateTensor<bool>(model.input3(), {true, true, true, true}); in TEST()
77 model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); in TEST()
78 model.PopulateTensor<float>(model.input2(), {0.1, 0.2, 0.3, 0.4}); in TEST()
79 model.PopulateTensor<float>(model.input3(), {0.5, 0.6, 0.7, 0.8}); in TEST()
90 model.PopulateTensor<bool>(model.input1(), {false, true, false, false}); in TEST()
91 model.PopulateTensor<uint8_t>(model.input2(), {1, 2, 3, 4}); in TEST()
92 model.PopulateTensor<uint8_t>(model.input3(), {5, 6, 7, 8}); in TEST()
103 model.PopulateTensor<bool>(model.input1(), {false, true, false, false}); in TEST()
[all …]
Drange_test.cc55 model.PopulateTensor<int32_t>(model.start(), {0}); in TEST()
56 model.PopulateTensor<int32_t>(model.limit(), {4}); in TEST()
57 model.PopulateTensor<int32_t>(model.delta(), {1}); in TEST()
65 model.PopulateTensor<int32_t>(model.start(), {2}); in TEST()
66 model.PopulateTensor<int32_t>(model.limit(), {9}); in TEST()
67 model.PopulateTensor<int32_t>(model.delta(), {2}); in TEST()
75 model.PopulateTensor<int32_t>(model.start(), {10}); in TEST()
76 model.PopulateTensor<int32_t>(model.limit(), {3}); in TEST()
77 model.PopulateTensor<int32_t>(model.delta(), {-3}); in TEST()
85 model.PopulateTensor<float>(model.start(), {0}); in TEST()
[all …]
Dsparse_to_dense_test.cc47 PopulateTensor<T>(default_value_, {default_value}); in SparseToDenseOpModel()
68 m.PopulateTensor<int32_t>(m.indices(), {3}); in TEST()
69 m.PopulateTensor<int32_t>(m.output_shape(), {5}); in TEST()
70 m.PopulateTensor<float>(m.values(), {7}); in TEST()
80 m.PopulateTensor<int32_t>(m.indices(), {1, 3, 5}); in TEST()
81 m.PopulateTensor<int32_t>(m.output_shape(), {7}); in TEST()
82 m.PopulateTensor<float>(m.values(), {2, 4, 6}); in TEST()
92 m.PopulateTensor<int32_t>(m.indices(), {0, 0, 0, 1, 2, 1, 2, 0, 1}); in TEST()
93 m.PopulateTensor<int32_t>(m.output_shape(), {3, 3, 3}); in TEST()
94 m.PopulateTensor<float>(m.values(), {2, 4, 6}); in TEST()
[all …]
Dreverse_sequence_test.cc58 model.PopulateTensor<float>(model.input(), in TEST()
61 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 2, 3, 3}); in TEST()
73 model.PopulateTensor<float>(model.input(), in TEST()
76 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 4}); in TEST()
88 model.PopulateTensor<int32_t>( in TEST()
91 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 2, 3, 3}); in TEST()
103 model.PopulateTensor<int32_t>( in TEST()
106 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 4}); in TEST()
118 model.PopulateTensor<int64_t>( in TEST()
121 model.PopulateTensor<int32_t>(model.seq_lengths(), {3, 2, 3, 3}); in TEST()
[all …]
Dreverse_test.cc56 model.PopulateTensor<float>(model.input(), {1, 2, 3, 4}); in TEST()
57 model.PopulateTensor<int32_t>(model.axis(), {0}); in TEST()
67 model.PopulateTensor<float>(model.input(), in TEST()
70 model.PopulateTensor<int32_t>(model.axis(), {1}); in TEST()
84 model.PopulateTensor<int32_t>(model.input(), {1, 2, 3, 4}); in TEST()
85 model.PopulateTensor<int32_t>(model.axis(), {0}); in TEST()
95 model.PopulateTensor<int32_t>( in TEST()
98 model.PopulateTensor<int32_t>(model.axis(), {1}); in TEST()
112 model.PopulateTensor<int64_t>(model.input(), {1, 2, 3, 4}); in TEST()
113 model.PopulateTensor<int32_t>(model.axis(), {0}); in TEST()
[all …]
Darg_min_max_test.cc81 model.PopulateTensor<float>(model.input(), {0.1, 0.9, 0.7, 0.3}); in TEST()
82 model.PopulateTensor<int>(model.axis(), {3}); in TEST()
92 model.PopulateTensor<uint8_t>(model.input(), {1, 9, 7, 3}); in TEST()
93 model.PopulateTensor<int>(model.axis(), {3}); in TEST()
103 model.PopulateTensor<int8_t>(model.input(), {-1, -9, 7, 3}); in TEST()
104 model.PopulateTensor<int>(model.axis(), {3}); in TEST()
114 model.PopulateTensor<int>(model.input(), {1, 9, 7, 3}); in TEST()
115 model.PopulateTensor<int>(model.axis(), {3}); in TEST()
125 model.PopulateTensor<int>(model.input(), {1, 2, 7, 8, 1, 9, 7, 3}); in TEST()
126 model.PopulateTensor<int>(model.axis(), {3}); in TEST()
[all …]
Dcomparisons_test.cc98 model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); in TEST()
99 model.PopulateTensor<bool>(model.input2(), {true, true, false, false}); in TEST()
109 model.PopulateTensor<float>(model.input1(), {0.1, 0.9, 0.7, 0.3}); in TEST()
110 model.PopulateTensor<float>(model.input2(), {0.1, 0.2, 0.6, 0.5}); in TEST()
120 model.PopulateTensor<int>(model.input1(), {-1, 9, 7, 3}); in TEST()
121 model.PopulateTensor<int>(model.input2(), {1, 2, 7, 5}); in TEST()
131 model.PopulateTensor<int>(model.input1(), {-1, 9, 7, 3}); in TEST()
132 model.PopulateTensor<int>(model.input2(), {7}); in TEST()
142 model.PopulateTensor<int>(model.input1(), {-1, 9, 7, 3, 2, 4, 2, 8}); in TEST()
143 model.PopulateTensor<int>(model.input2(), {7, 1, 2, 4}); in TEST()
[all …]
Doptional_tensor_test.cc114 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights()
118 PopulateTensor(input_to_forget_weights_, f); in SetInputToForgetWeights()
122 PopulateTensor(input_to_cell_weights_, f); in SetInputToCellWeights()
126 PopulateTensor(input_to_output_weights_, f); in SetInputToOutputWeights()
130 PopulateTensor(recurrent_to_input_weights_, f); in SetRecurrentToInputWeights()
134 PopulateTensor(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
138 PopulateTensor(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
142 PopulateTensor(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
146 PopulateTensor(cell_to_input_weights_, f); in SetCellToInputWeights()
150 PopulateTensor(cell_to_forget_weights_, f); in SetCellToForgetWeights()
[all …]
Ddiv_test.cc66 m.PopulateTensor<float>(m.input1(), {-0.2, 0.2, -1.2, 0.8}); in TEST()
67 m.PopulateTensor<float>(m.input2(), {0.5, 0.2, -1.5, 0.5}); in TEST()
77 m.PopulateTensor<float>(m.input1(), {-0.2, 0.2, -1.2, 0.8}); in TEST()
78 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, -1.5, 0.5}); in TEST()
91 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.3, 0.8, 1.1, -2.0}); in TEST()
92 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.6, 0.5, -1.1, -0.1}); in TEST()
108 m.PopulateTensor<float>(m.input1(), {-0.2, 0.2, 0.07, 0.08, 0.11, -0.123}); in TEST()
109 m.PopulateTensor<float>(m.input2(), {0.1}); in TEST()
122 m.PopulateTensor<int32_t>(m.input1(), {-2, 2, -15, 8}); in TEST()
123 m.PopulateTensor<int32_t>(m.input2(), {5, -2, -3, 5}); in TEST()
[all …]
Dfloor_mod_test.cc55 model.PopulateTensor<int32_t>(model.input1(), {10, 9, 11, 3}); in TEST()
56 model.PopulateTensor<int32_t>(model.input2(), {2, 2, 3, 4}); in TEST()
66 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
67 model.PopulateTensor<int32_t>(model.input2(), {2, 2, -3, -4}); in TEST()
76 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
77 model.PopulateTensor<int32_t>(model.input2(), {-3}); in TEST()
86 model.PopulateTensor<int64_t>(model.input1(), {10, -9, -11, (1LL << 34) + 9}); in TEST()
87 model.PopulateTensor<int64_t>(model.input2(), {-(1LL << 33)}); in TEST()
98 model.PopulateTensor<float>(model.input1(), {10, 9, 11, 3}); in TEST()
99 model.PopulateTensor<float>(model.input2(), {2, 2, 3, 4}); in TEST()
[all …]
Dsquared_difference_test.cc67 m.PopulateTensor<float>(m.input1(), {-0.2, 0.2, -1.2, 0.8}); in TEST()
68 m.PopulateTensor<float>(m.input2(), {0.5, 0.2, -1.5, 0.5}); in TEST()
81 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.3, 0.8, 1.1, -2.0}); in TEST()
82 m.PopulateTensor<float>(m.input2(), {1.0, 0.2, 0.6, 0.4, -1.0, -0.0}); in TEST()
99 m.PopulateTensor<float>(m.input1(), {-0.2, 0.2, 0.5, 0.8, 0.11, 1.1}); in TEST()
100 m.PopulateTensor<float>(m.input2(), {0.1}); in TEST()
113 m.PopulateTensor<int32_t>(m.input1(), {-2, 2, -15, 8}); in TEST()
114 m.PopulateTensor<int32_t>(m.input2(), {5, -2, -3, 5}); in TEST()
126 m.PopulateTensor<int32_t>(m.input1(), {-20, 2, 3, 8, 11, -20}); in TEST()
127 m.PopulateTensor<int32_t>(m.input2(), {1, 2, 6, 5, -5, -20}); in TEST()
[all …]
Dpow_test.cc56 model.PopulateTensor<int32_t>(model.input1(), {12, 2, 7, 8}); in TEST()
57 model.PopulateTensor<int32_t>(model.input2(), {1, 2, 3, 1}); in TEST()
67 model.PopulateTensor<int32_t>(model.input1(), {0, 2, -7, 8}); in TEST()
68 model.PopulateTensor<int32_t>(model.input2(), {1, 2, 3, 0}); in TEST()
78 model.PopulateTensor<float>(model.input1(), {0.3, 0.4, 0.7, 5.8}); in TEST()
79 model.PopulateTensor<float>(model.input2(), {0.5, 2.7, 3.1, 3.2}); in TEST()
91 model.PopulateTensor<float>(model.input1(), {0.3, 0.4, 0.7, 5.8}); in TEST()
92 model.PopulateTensor<float>(model.input2(), {0.5, -2.7, 3.1, -3.2}); in TEST()
103 model.PopulateTensor<int32_t>(model.input1(), {12, 2, 7, 8}); in TEST()
104 model.PopulateTensor<int32_t>(model.input2(), {4}); in TEST()
Dwhere_test.cc54 m.PopulateTensor<bool>(m.input(), {false, false, false}); in TEST()
61 m.PopulateTensor<bool>(m.input(), {true, false, true}); in TEST()
68 m.PopulateTensor<bool>(m.input(), {false, false, false, // in TEST()
77 m.PopulateTensor<bool>(m.input(), {true, false, true}); in TEST()
85 m.PopulateTensor<bool>(m.input(), {true, true, false, // in TEST()
98 m.PopulateTensor<bool>(m.input(), {true, false, false, true, true, // in TEST()
113 m.PopulateTensor<bool>(m.input(), {false, false, false, false, // in TEST()
121 m.PopulateTensor<bool>(m.input(), {true, false, true, // in TEST()
131 m.PopulateTensor<bool>(m.input(), {true, true, false, true, // in TEST()
143 m.PopulateTensor<bool>(m.input(), {true, true, false, true, false, false, // in TEST()
Dmul_test.cc92 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
93 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5}); in TEST()
103 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
104 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 5}); in TEST()
117 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0}); in TEST()
118 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5, 1.1, 0.1}); in TEST()
134 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0}); in TEST()
135 m.PopulateTensor<float>(m.input2(), {0.1}); in TEST()
148 m.PopulateTensor<int32_t>(m.input1(), {-20, 2, 7, 8}); in TEST()
149 m.PopulateTensor<int32_t>(m.input2(), {1, 2, 3, 5}); in TEST()
[all …]
Dlogical_test.cc67 model.PopulateTensor<bool>(model.input1(), {true, false, false, true}); in TEST()
68 model.PopulateTensor<bool>(model.input2(), {true, false, true, false}); in TEST()
77 model.PopulateTensor<bool>(model.input1(), {true, false, false, true}); in TEST()
78 model.PopulateTensor<bool>(model.input2(), {false}); in TEST()
87 model.PopulateTensor<bool>(model.input1(), {true, false, false, true}); in TEST()
88 model.PopulateTensor<bool>(model.input2(), {true, false, true, false}); in TEST()
97 model.PopulateTensor<bool>(model.input1(), {true, false, false, true}); in TEST()
98 model.PopulateTensor<bool>(model.input2(), {true}); in TEST()
Dcast_test.cc48 m.PopulateTensor<int64_t>(m.input(), {100, 200, 300, 400, 500, 600}); in TEST()
56 m.PopulateTensor<float>(m.input(), {100.f, 20.f, 3.f, 0.4f, 0.999f, 1.1f}); in TEST()
64 m.PopulateTensor<float>(m.input(), {100.f, -1.0f, 0.f, 0.4f, 0.999f, 1.1f}); in TEST()
72 m.PopulateTensor<bool>(m.input(), {true, true, false, true, false, true}); in TEST()
80 m.PopulateTensor<std::complex<float>>( in TEST()
92 m.PopulateTensor<float>(m.input(), {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f}); in TEST()
104 m.PopulateTensor<std::complex<float>>( in TEST()
116 m.PopulateTensor<int>(m.input(), {1, 2, 3, 4, 5, 6}); in TEST()
128 m.PopulateTensor<std::complex<float>>( in TEST()
Dfill_test.cc50 m.PopulateTensor<int32_t>(m.input1(), {2, 3}); in TEST()
51 m.PopulateTensor<int32_t>(m.input2(), {-11}); in TEST()
60 m.PopulateTensor<int32_t>(m.input1(), {2, 4}); in TEST()
61 m.PopulateTensor<int64_t>(m.input2(), {2 ^ 45}); in TEST()
71 m.PopulateTensor<int64_t>(m.input1(), {2, 2, 2}); in TEST()
72 m.PopulateTensor<float>(m.input2(), {4.0}); in TEST()
81 m.PopulateTensor<float>(m.input2(), {4.0}); in TEST()
Delementwise_test.cc61 m.PopulateTensor<float>(m.input(), {0, 3.1415926, -3.1415926, 1}); in TEST()
70 m.PopulateTensor<float>(m.input(), {0, 3.1415926, -3.1415926, 1}); in TEST()
79 m.PopulateTensor<float>(m.input(), {1, 3.1415926, 1, 1}); in TEST()
88 m.PopulateTensor<float>(m.input(), { in TEST()
101 m.PopulateTensor<float>(m.input(), {0, 1, 2, 4}); in TEST()
110 m.PopulateTensor<float>(m.input(), {1, 2, 4, 9}); in TEST()
119 m.PopulateTensor<float>(m.input(), {1, 2, 0.5, -3.0}); in TEST()
128 m.PopulateTensor<bool>(m.input(), {true, false, true, false}); in TEST()
Dadd_test.cc93 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
94 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5}); in TEST()
103 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
104 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5}); in TEST()
116 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0}); in TEST()
117 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5, 1.1, 0.1}); in TEST()
132 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8, 1.1, 2.0}); in TEST()
133 m.PopulateTensor<float>(m.input2(), {0.1}); in TEST()
146 m.PopulateTensor<int32_t>(m.input1(), {-20, 2, 7, 8}); in TEST()
147 m.PopulateTensor<int32_t>(m.input2(), {1, 2, 3, 5}); in TEST()
[all …]
Dadd_n_test.cc72 m.PopulateTensor<float>(m.input(0), {-2.0, 0.2, 0.7, 0.8}); in TEST()
73 m.PopulateTensor<float>(m.input(1), {0.1, 0.2, 0.3, 0.5}); in TEST()
74 m.PopulateTensor<float>(m.input(2), {0.5, 0.1, 0.1, 0.2}); in TEST()
84 m.PopulateTensor<int32_t>(m.input(0), {-20, 2, 7, 8}); in TEST()
85 m.PopulateTensor<int32_t>(m.input(1), {1, 2, 3, 5}); in TEST()
86 m.PopulateTensor<int32_t>(m.input(2), {10, -5, 1, -2}); in TEST()
Dfloor_div_test.cc55 model.PopulateTensor<int32_t>(model.input1(), {10, 9, 11, 3}); in TEST()
56 model.PopulateTensor<int32_t>(model.input2(), {2, 2, 3, 4}); in TEST()
66 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
67 model.PopulateTensor<int32_t>(model.input2(), {2, 2, -3, -4}); in TEST()
76 model.PopulateTensor<int32_t>(model.input1(), {10, -9, -11, 7}); in TEST()
77 model.PopulateTensor<int32_t>(model.input2(), {-3}); in TEST()
Dsub_test.cc93 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 1.7, 0.5}); in TEST()
94 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.8}); in TEST()
104 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 1.7, 0.5}); in TEST()
105 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.8}); in TEST()
118 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 1.7, 0.5, -1.1, 2.0}); in TEST()
119 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.8, -1.1, 0.1}); in TEST()
135 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 1.7, 0.5, -1.1, 2.0}); in TEST()
136 m.PopulateTensor<float>(m.input2(), {0.5}); in TEST()
149 m.PopulateTensor<int32_t>(m.input1(), {-20, 2, 7, 8}); in TEST()
150 m.PopulateTensor<int32_t>(m.input2(), {1, 2, 3, 5}); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc71 PopulateTensor(index, data); in SetData()
120 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
121 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5}); in TEST()
133 m.PopulateTensor<float>(m.input1(), {-2.0, -1.0, 1.0, 2.0}); in TEST()
134 m.PopulateTensor<float>(m.input2(), {1.0, 2.0, 3.0, 4.0}); in TEST()
144 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
145 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5}); in TEST()
156 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
157 m.PopulateTensor<float>(m.input2(), {0.1, 0.2, 0.3, 0.5}); in TEST()
189 m.PopulateTensor<float>(m.input1(), {-2.0, 0.2, 0.7, 0.8}); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder_test.cc99 m.PopulateTensor<float>(m.inputs(), in TEST()
101 m.PopulateTensor<int>(m.sequence_length(), {2}); in TEST()
125 m.PopulateTensor<float>( in TEST()
133 m.PopulateTensor<int>(m.sequence_length(), {3, 3, 3}); in TEST()
157 m.PopulateTensor<float>( in TEST()
165 m.PopulateTensor<int>(m.sequence_length(), {3, 3}); in TEST()
196 m.PopulateTensor<float>( in TEST()
206 m.PopulateTensor<int>(m.sequence_length(), {1, 2, 3}); in TEST()

1234