Home
last modified time | relevance | path

Searched refs:IgniteDataset (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/
Dignite_dataset.cc21 IgniteDataset::IgniteDataset(OpKernelContext* ctx, string cache_name, in IgniteDataset() function in tensorflow::IgniteDataset
53 IgniteDataset::~IgniteDataset() { LOG(INFO) << "Ignite Dataset destroyed"; } in ~IgniteDataset()
55 std::unique_ptr<IteratorBase> IgniteDataset::MakeIteratorInternal( in MakeIteratorInternal()
66 const DataTypeVector& IgniteDataset::output_dtypes() const { return dtypes_; } in output_dtypes()
68 const std::vector<PartialTensorShape>& IgniteDataset::output_shapes() const { in output_shapes()
72 string IgniteDataset::DebugString() const { return "IgniteDatasetOp::Dataset"; } in DebugString()
74 Status IgniteDataset::AsGraphDefInternal(SerializationContext* ctx, in AsGraphDefInternal()
Dignite_dataset.h23 class IgniteDataset : public DatasetBase {
25 IgniteDataset(OpKernelContext* ctx, string cache_name, string host,
31 ~IgniteDataset();
Dignite_dataset_iterator.h26 class IgniteDatasetIterator : public DatasetIterator<IgniteDataset> {
Dignite_dataset_ops.cc185 *output = new IgniteDataset( in MakeDataset()
Dignite_dataset_iterator.cc30 : DatasetIterator<IgniteDataset>(params), in IgniteDatasetIterator()
/external/tensorflow/tensorflow/contrib/ignite/
DREADME.md57 >>> from tensorflow.contrib.ignite import IgniteDataset
60 >>> dataset = IgniteDataset(cache_name="SQL_PUBLIC_KITTEN_CACHE")
75 >>> from tensorflow.contrib.ignite import IgniteDataset
78 >>> dataset = IgniteDataset(cache_name="IMAGES")
100 >>> from tensorflow.contrib.ignite import IgniteDataset
103 >>> dataset = IgniteDataset(cache_name="IMAGES").map(lambda obj: obj['val']['pixels'])
133 >>> from tensorflow.contrib.ignite import IgniteDataset
135 >>> dataset = IgniteDataset("IMAGES")
185 >>> from tensorflow.contrib.ignite import IgniteDataset
188 >>> dataset = IgniteDataset(cache_name="IMAGES",
D__init__.py35 from tensorflow.contrib.ignite.python.ops.ignite_dataset_ops import IgniteDataset
/external/tensorflow/tensorflow/contrib/ignite/python/tests/
Dignite_dataset_test.py23 from tensorflow.contrib.ignite import IgniteDataset
47 ds = IgniteDataset(cache_name="SQL_PUBLIC_TEST_CACHE", port=42300)
/external/tensorflow/tensorflow/contrib/ignite/python/ops/
Dignite_dataset_ops.py690 class IgniteDataset(dataset_ops.DatasetSource): class
761 super(IgniteDataset, self).__init__(self._as_variant_tensor())