Searched refs:TakeDataset (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | take_dataset_op.cc | 35 TakeDataset::TakeDataset(OpKernelContext* ctx, int64 count, in TakeDataset() function in tensorflow::data::TakeDataset
41 TakeDataset::TakeDataset(DatasetContext::Params params, int64 count, in TakeDataset() function in tensorflow::data::TakeDataset
49 TakeDataset::~TakeDataset() { input_->Unref(); } in ~TakeDataset()
51 const DataTypeVector& TakeDataset::output_dtypes() const { in output_dtypes()
55 const std::vector<PartialTensorShape>& TakeDataset::output_shapes() const { in output_shapes()
59 string TakeDataset::DebugString() const { in DebugString()
63 int64 TakeDataset::Cardinality() const { in Cardinality()
77 Status TakeDataset::InputDatasets( in InputDatasets()
83 Status TakeDataset::CheckExternalState() const { in CheckExternalState()
87 class TakeDataset::EmptyIterator : public DatasetIterator<TakeDataset> {
[all …]
|
| D | take_dataset_op.h | 23 class TakeDataset : public DatasetBase {
25 TakeDataset(OpKernelContext* ctx, int64 count, const DatasetBase* input);
27 TakeDataset(DatasetContext::Params params, int64 count,
30 ~TakeDataset() override;
|
| /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/ops/compat/ops_history_v2/ |
| D | TakeDataset.pbtxt | 2 name: "TakeDataset"
30 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TakeDataset.pbtxt | 2 name: "TakeDataset"
30 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TakeDataset.pbtxt | 2 graph_op_name: "TakeDataset"
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_branch_dataset_op.cc | 528 temp_dataset = new TakeDataset(std::move(take_dataset_params), count, in MakeCurrentIterator()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 1420 return TakeDataset(self, count)
3908 class TakeDataset(UnaryUnchangedStructureDataset): class
3919 super(TakeDataset, self).__init__(input_dataset, variant_tensor)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4828 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4828 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 81612 name: "TakeDataset"
81640 name: "TakeDataset"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 21091 func TakeDataset(scope *Scope, input_dataset tf.Output, count tf.Output, output_types []tf.DataType… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 54112 name: "TakeDataset"
|