Searched refs:CopySliceData (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | save_op_test.cc | 149 EXPECT_TRUE(reader.CopySliceData("tensor_bool", s, data)); in TEST_F() 168 EXPECT_TRUE(reader.CopySliceData("tensor_int", s, data)); in TEST_F() 187 EXPECT_TRUE(reader.CopySliceData("tensor_float", s, data)); in TEST_F() 206 EXPECT_TRUE(reader.CopySliceData("tensor_double", s, data)); in TEST_F() 224 EXPECT_TRUE(reader.CopySliceData("tensor_qint8", s, data)); in TEST_F() 242 EXPECT_TRUE(reader.CopySliceData("tensor_qint32", s, data)); in TEST_F() 260 EXPECT_TRUE(reader.CopySliceData("tensor_uint8", s, data)); in TEST_F() 278 EXPECT_TRUE(reader.CopySliceData("tensor_int8", s, data)); in TEST_F() 296 EXPECT_TRUE(reader.CopySliceData("tensor_int16", s, data)); in TEST_F() 314 EXPECT_TRUE(reader.CopySliceData("tensor_int64", s, data)); in TEST_F() [all …]
|
D | save_restore_tensor.cc | 220 reader->CopySliceData(tensor_name, slice_to_load, \ in RestoreTensor()
|
/external/tensorflow/tensorflow/core/util/ |
D | tensor_slice_reader_test.cc | 141 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleFloatHelper() 156 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleFloatHelper() 170 EXPECT_FALSE(reader.CopySliceData("test", s, results)); in SimpleFloatHelper() 266 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleIntXHelper() 281 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleIntXHelper() 295 EXPECT_FALSE(reader.CopySliceData("test", s, results)); in SimpleIntXHelper()
|
D | tensor_slice_reader.h | 91 bool CopySliceData(const string& name, const TensorSlice& slice, 145 bool TensorSliceReader::CopySliceData(const string& name, in CopySliceData() function
|
D | tensor_slice_reader.cc | 251 success = CopySliceData(name, slice, \ in GetTensor()
|