Searched refs:TakeDataset (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | take_dataset_op.cc | 35 *output = new TakeDataset(ctx, count, input); in MakeDataset()
42 class TakeDataset::EmptyIterator : public DatasetIterator<TakeDataset> {
45 : DatasetIterator<TakeDataset>(params) {} in EmptyIterator()
69 class TakeDataset::FiniteIterator : public DatasetIterator<TakeDataset> {
72 : DatasetIterator<TakeDataset>(params), i_(0) {} in FiniteIterator()
138 std::unique_ptr<IteratorBase> TakeDataset::MakeIteratorInternal( in MakeIteratorInternal()
149 Status TakeDataset::AsGraphDefInternal(SerializationContext* ctx, in AsGraphDefInternal()
|
| D | take_dataset_op.h | 25 class TakeDataset : public DatasetBase {
27 TakeDataset(OpKernelContext* ctx, int64 count, const DatasetBase* input) in TakeDataset() function
32 TakeDataset(DatasetContext::Params params, int64 count, in TakeDataset() function
40 ~TakeDataset() override { input_->Unref(); } in ~TakeDataset()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_TakeDataset.pbtxt | 2 graph_op_name: "TakeDataset"
4 name: "data.TakeDataset"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TakeDataset.pbtxt | 2 graph_op_name: "TakeDataset"
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils.py | 1188 dataset_ops.TakeDataset,
1244 dataset_ops.TakeDataset,
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_branch_dataset_op.cc | 501 temp_dataset = new TakeDataset(std::move(take_dataset_params), count, in MakeCurrentIterator()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 778 return TakeDataset(self, count)
2713 class TakeDataset(UnaryUnchangedStructureDataset): class
2724 super(TakeDataset, self).__init__(input_dataset, variant_tensor)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3864 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3864 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 20600 func TakeDataset(scope *Scope, input_dataset tf.Output, count tf.Output, output_types []tf.DataType… func
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 83119 name: "TakeDataset"
83147 name: "TakeDataset"
|
| D | ops_history.v2.pbtxt | 82492 name: "TakeDataset"
82520 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 40108 name: "TakeDataset"
|