Home
last modified time | relevance | path

Searched refs:arg0_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc72 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
83 add.arg0_data()[0] = 123; in TEST()
94 EXPECT_EQ(add_const.arg0_data()[0], 123); in TEST()
95 EXPECT_EQ(add_const.arg0_data(), add.arg_data(0)); in TEST()
114 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
126 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
135 add.arg0_data()[0] = 111; in TEST()
145 EXPECT_EQ(add_const.arg0_data()[0], 111); in TEST()
146 EXPECT_EQ(add_const.arg0_data(), add_const.arg_data(0)); in TEST()
154 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen_test_h.golden127 float* arg0_data() {
134 const float* arg0_data() const {
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Dtfcompile.md173 float* arg0_data();
232 std::copy(args + 0, args + 6, matmul.arg0_data());
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc4478 auto arg0_data = args[0].data<uint32>(); in TEST_F() local
4480 std::vector<uint32> expected_data = {arg0_data[0] + arg1_data[0]}; in TEST_F()