Searched refs:arg0_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 51 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
62 add.arg0_data()[0] = 123; in TEST()
73 EXPECT_EQ(add_const.arg0_data()[0], 123); in TEST()
74 EXPECT_EQ(add_const.arg0_data(), add.arg_data(0)); in TEST()
92 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
104 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
113 add.arg0_data()[0] = 111; in TEST()
123 EXPECT_EQ(add_const.arg0_data()[0], 111); in TEST()
124 EXPECT_EQ(add_const.arg0_data(), add_const.arg_data(0)); in TEST()
132 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test_h.golden | 121 float* arg0_data() {
128 const float* arg0_data() const {
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | tfcompile.md | 175 float* arg0_data();
234 std::copy(args + 0, args + 6, matmul.arg0_data());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 3251 auto arg0_data = args[0].data<uint32>(); in TEST_F() local
3253 std::vector<uint32> expected_data = {arg0_data[0] + arg1_data[0]}; in TEST_F()
|