Searched refs:arg_data (Results 1 – 7 of 7) 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()
52 EXPECT_EQ(add.arg1_data(), add.arg_data(1)); in TEST()
74 EXPECT_EQ(add_const.arg0_data(), add.arg_data(0)); in TEST()
77 EXPECT_EQ(add_const.arg1_data(), add.arg_data(1)); in TEST()
92 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); in TEST()
93 EXPECT_EQ(add.arg1_data(), add.arg_data(1)); in TEST()
104 EXPECT_EQ(add.arg0_data(), add.arg_data(0)); 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()
152 EXPECT_EQ(add_const.arg0_data(), add_const.arg_data(0)); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convert_test.cc | 215 std::unique_ptr<GlobalData> arg_data = in XLA_TEST_F() local
224 ComputeAndCompareR1<float>(&builder, expected, {arg_data.get()}); in XLA_TEST_F()
234 std::unique_ptr<GlobalData> arg_data = in XLA_TEST_F() local
243 ComputeAndCompareR1<float>(&builder, expected, {arg_data.get()}); in XLA_TEST_F()
252 std::unique_ptr<GlobalData> arg_data = in XLA_TEST_F() local
261 ComputeAndCompareR1<uint32>(&builder, expected, {arg_data.get()}); in XLA_TEST_F()
269 std::unique_ptr<GlobalData> arg_data = in XLA_TEST_F() local
278 ComputeAndCompareR1<int64>(&builder, expected, {arg_data.get()}); in XLA_TEST_F()
286 std::unique_ptr<GlobalData> arg_data = in XLA_TEST_F() local
295 ComputeAndCompareR1<int64>(&builder, expected, {arg_data.get()}); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test_h.golden | 122 return static_cast<float*>(arg_data(0));
126 arg_data(0)))[dim0][dim1];
129 return static_cast<const float*>(arg_data(0));
133 arg_data(0)))[dim0][dim1];
140 return static_cast<float*>(arg_data(0));
144 arg_data(0)))[dim0][dim1];
147 return static_cast<const float*>(arg_data(0));
151 arg_data(0)))[dim0][dim1];
158 return static_cast<tensorflow::int64*>(arg_data(1));
162 arg_data(1)))[dim0][dim1];
[all …]
|
| D | test.cc | 56 memset(computation->arg_data(i), 0, computation->arg_size(i)); in zero_buffers()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 84 *static_cast<int32*>(function.arg_data(0)) = 10; in TEST()
85 *static_cast<int32*>(function.arg_data(1)) = 32; in TEST()
91 *static_cast<int32*>(function.arg_data(0)) = 100; in TEST()
92 *static_cast<int32*>(function.arg_data(1)) = 320; in TEST()
|
| D | xla_compiled_cpu_function.h | 143 void* arg_data(size_t index) { in arg_data() function
146 const void* arg_data(size_t index) const { in arg_data() function
|
| /external/syzkaller/executor/ |
| D | executor.cc | 139 const uint64 arg_data = 2; variable
562 case arg_data: { in execute_one()
|