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.cc149 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 …]
Dsave_restore_tensor.cc220 reader->CopySliceData(tensor_name, slice_to_load, \ in RestoreTensor()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader_test.cc141 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()
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.cc251 success = CopySliceData(name, slice, \ in GetTensor()