Searched refs:src_tensor_1 (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | elementwise_test_util.cc | 502 TensorFloat32 src_tensor_0, src_tensor_1; in SubTest() local
504 src_tensor_1.shape = BHWC(1, 2, 1, 2); in SubTest()
506 src_tensor_1.data = {0.5f, 1.0f, 3.0f, 3.5f}; in SubTest()
519 op_def, OperationType::SUB, src_tensor_1.shape); in SubTest()
521 {src_tensor_0, src_tensor_1}, in SubTest()
532 TensorFloat32 src_tensor_0, src_tensor_1; in SquaredDiffTest() local
534 src_tensor_1.shape = BHWC(1, 2, 1, 2); in SquaredDiffTest()
536 src_tensor_1.data = {0.5f, 1.0f, 3.0f, 3.5f}; in SquaredDiffTest()
549 op_def, OperationType::SQUARED_DIFF, src_tensor_1.shape); in SquaredDiffTest()
551 {src_tensor_0, src_tensor_1}, in SquaredDiffTest()
[all …]
|