Searched refs:arg1_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 52 EXPECT_EQ(add.arg1_data(), add.arg_data(1)); in TEST()
63 add.arg1_data()[0] = 456; in TEST()
76 EXPECT_EQ(add_const.arg1_data()[0], 456); in TEST()
77 EXPECT_EQ(add_const.arg1_data(), add.arg_data(1)); in TEST()
93 EXPECT_EQ(add.arg1_data(), add.arg_data(1)); in TEST()
161 EXPECT_EQ(cond.arg1_data(), cond.arg_data(1)); in TEST()
186 EXPECT_EQ(gather.arg1_data(), gather.arg_data(1)); in TEST()
193 std::copy(indices + 0, indices + 2, gather.arg1_data()); in TEST()
212 EXPECT_EQ(gather_const.arg1_data()[i], indices[i]); in TEST()
214 EXPECT_EQ(gather_const.arg1_data(), gather_const.arg_data(1)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test_h.golden | 157 tensorflow::int64* arg1_data() {
164 const tensorflow::int64* arg1_data() const {
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | tfcompile.md | 179 float* arg1_data();
235 std::copy(args + 6, args + 12, matmul.arg1_data());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 3252 auto arg1_data = args[1].data<uint32>(); in TEST_F() local
3253 std::vector<uint32> expected_data = {arg0_data[0] + arg1_data[0]}; in TEST_F()
|