HomeSort by relevance Sort by last modified time
    Searched refs:HasTensor (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
save_op_test.cc 140 EXPECT_TRUE(reader.HasTensor("tensor_bool", &shape, &type));
159 EXPECT_TRUE(reader.HasTensor("tensor_int", &shape, &type));
178 EXPECT_TRUE(reader.HasTensor("tensor_float", &shape, &type));
197 EXPECT_TRUE(reader.HasTensor("tensor_double", &shape, &type));
216 EXPECT_TRUE(reader.HasTensor("tensor_qint8", &shape, &type));
234 EXPECT_TRUE(reader.HasTensor("tensor_qint32", &shape, &type));
252 EXPECT_TRUE(reader.HasTensor("tensor_uint8", &shape, &type));
270 EXPECT_TRUE(reader.HasTensor("tensor_int8", &shape, &type));
288 EXPECT_TRUE(reader.HasTensor("tensor_int16", &shape, &type));
306 EXPECT_TRUE(reader.HasTensor("tensor_int64", &shape, &type))
    [all...]
save_restore_tensor.cc 176 context, reader->HasTensor(tensor_name, &saved_shape, &type),
  /external/tensorflow/tensorflow/c/
checkpoint_reader.h 44 bool HasTensor(const string& name) const;
checkpoint_reader.cc 65 bool CheckpointReader::HasTensor(const string& name) const {
67 return reader_->HasTensor(name, nullptr, nullptr);
  /external/tensorflow/tensorflow/core/util/
tensor_slice_reader_test.cc 123 EXPECT_TRUE(reader.HasTensor("test", &shape, &type));
126 EXPECT_FALSE(reader.HasTensor("don't exist", nullptr, nullptr));
248 EXPECT_TRUE(reader.HasTensor("test", &shape, &type));
251 EXPECT_FALSE(reader.HasTensor("don't exist", nullptr, nullptr));
382 EXPECT_TRUE(reader->HasTensor("test", &shape, &type));
385 EXPECT_FALSE(reader->HasTensor("don't exist", nullptr, nullptr));
tensor_slice_reader.h 84 bool HasTensor(const string& name, TensorShape* shape, DataType* type) const;
tensor_slice_reader.cc 201 bool TensorSliceReader::HasTensor(const string& name, TensorShape* shape,
  /external/tensorflow/tensorflow/python/util/
py_checkpoint_reader.i 138 %rename("_HasTensor") tensorflow::checkpoint::CheckpointReader::HasTensor;

Completed in 868 milliseconds