Searched refs:expected_values_tensor (Results 1 – 3 of 3) sorted by relevance
123 const Tensor& expected_values_tensor = outputs[0]; in TestMulShape() local125 expected_values_tensor.flat<float>().data(); in TestMulShape()128 expected_values_data + expected_values_tensor.NumElements()); in TestMulShape()130 for (const int64 dim : expected_values_tensor.shape().dim_sizes()) { in TestMulShape()
123 const Tensor& expected_values_tensor = outputs[0]; in TestAddShape() local125 expected_values_tensor.flat<float>().data(); in TestAddShape()128 expected_values_data + expected_values_tensor.NumElements()); in TestAddShape()130 for (const int64 dim : expected_values_tensor.shape().dim_sizes()) { in TestAddShape()
220 Tensor expected_values_tensor(DT_FLOAT, TensorShape({3})); in TestSinglePartition() local221 test::FillValues<float>(&expected_values_tensor, {0.2, 1.2, 0.001}); in TestSinglePartition()223 expected_values_tensor, in TestSinglePartition()