Home
last modified time | relevance | path

Searched refs:CopySliceData (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsave_op_test.cc152 EXPECT_TRUE(reader.CopySliceData("tensor_bool", s, data)); in TEST_F()
171 EXPECT_TRUE(reader.CopySliceData("tensor_int", s, data)); in TEST_F()
190 EXPECT_TRUE(reader.CopySliceData("tensor_float", s, data)); in TEST_F()
209 EXPECT_TRUE(reader.CopySliceData("tensor_double", s, data)); in TEST_F()
227 EXPECT_TRUE(reader.CopySliceData("tensor_qint8", s, data)); in TEST_F()
245 EXPECT_TRUE(reader.CopySliceData("tensor_qint32", s, data)); in TEST_F()
263 EXPECT_TRUE(reader.CopySliceData("tensor_uint8", s, data)); in TEST_F()
281 EXPECT_TRUE(reader.CopySliceData("tensor_int8", s, data)); in TEST_F()
299 EXPECT_TRUE(reader.CopySliceData("tensor_int16", s, data)); in TEST_F()
317 EXPECT_TRUE(reader.CopySliceData("tensor_int64", s, data)); in TEST_F()
[all …]
Dsave_restore_tensor.cc234 reader->CopySliceData(tensor_name, slice_to_load, \ in RestoreTensor()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader_test.cc147 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleFloatHelper()
162 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleFloatHelper()
176 EXPECT_FALSE(reader.CopySliceData("test", s, results)); in SimpleFloatHelper()
272 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleIntXHelper()
287 EXPECT_TRUE(reader.CopySliceData("test", s, results)); in SimpleIntXHelper()
301 EXPECT_FALSE(reader.CopySliceData("test", s, results)); in SimpleIntXHelper()
Dtensor_slice_reader.h91 bool CopySliceData(const string& name, const TensorSlice& slice,
145 bool TensorSliceReader::CopySliceData(const string& name, in CopySliceData() function
Dtensor_slice_reader.cc263 success = CopySliceData(name, slice, \ in GetTensor()