Home
last modified time | relevance | path

Searched refs:FillWithYX (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc243 input.FillWithYX(Array2D<float>({ in TEST_F()
252 weights.FillWithYX(Array2D<float>({ in TEST_F()
261 expected.FillWithYX(Array2D<float>({ in TEST_F()
278 input.FillWithYX(Array2D<float>({ in TEST_F()
287 weights.FillWithYX(Array2D<float>({ in TEST_F()
296 expected.FillWithYX(Array2D<float>({ in TEST_F()
Darray4d.h111 void FillWithYX(const Array2D<T>& value) { in FillWithYX() function
Darray4d_test.cc156 array.FillWithYX(raster); in TEST()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc87 arhs->FillWithYX(rhs_raster); in RunTest()
124 input_data.FillWithYX(Array2D<T>({ in RunTest()
128 filter_data.FillWithYX(Array2D<T>({ in RunTest()
155 input_data.FillWithYX(Array2D<T>({ in RunTest()
162 filter_data.FillWithYX(Array2D<T>({ in RunTest()
189 input_data.FillWithYX(Array2D<T>({ in RunTest()
196 filter_data.FillWithYX(Array2D<T>({ in RunTest()
225 input_data.FillWithYX(Array2D<T>({{1.0f, 2.0f, 3.0f, 4.0f}, in RunTest()
230 filter_data.FillWithYX(Array2D<T>( in RunTest()
1560 input_data.FillWithYX(Array2D<bfloat16>({ in XLA_TEST_F()
[all …]
Dbroadcast_test.cc197 expected.FillWithYX(yx); in TEST_F()
261 expected.FillWithYX(to_broadcast); in TEST_F()
Dpad_test.cc149 input->FillWithYX(input_xy); in TEST_P()
278 input->FillWithYX(input_xy); in XLA_TEST_F()
Dreshape_test.cc521 t2x2x2x3.FillWithYX(*filler2x3); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1022 lhs_array.FillWithYX(Array2D<float>({ in TEST_P()
1035 rhs_array.FillWithYX(Array2D<float>({ in TEST_P()
1068 expected_array.FillWithYX(Array2D<float>({ in TEST_P()
1244 lhs_array.FillWithYX(Array2D<float>({ in TEST_P()
1257 rhs_array.FillWithYX(Array2D<float>({ in TEST_P()
1289 expected_array.FillWithYX(Array2D<float>({ in TEST_P()
1308 lhs_array.FillWithYX(Array2D<float>({ in TEST_P()
1321 rhs_array.FillWithYX(Array2D<float>({ in TEST_P()
1353 expected_array.FillWithYX(Array2D<float>({ in TEST_P()
1374 lhs_array.FillWithYX(Array2D<float>({ in TEST_P()
[all …]