Home
last modified time | relevance | path

Searched defs:output_dtypes (Results 1 – 25 of 99) sorted by relevance

1234

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dmap_defun.py28 output_dtypes, argument
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.h143 DataTypeVector output_dtypes() const { return output_dtypes_; } in output_dtypes() function
210 DataTypeVector output_dtypes, in BatchDatasetParams()
247 DataTypeVector type_arguments, DataTypeVector output_dtypes, in MapDatasetParams()
318 DataTypeVector output_dtypes, in TakeDatasetParams()
348 DataTypeVector output_dtypes, in ConcatenateDatasetParams()
Doptimize_dataset_op_test.cc29 DataTypeVector output_dtypes, in OptimizeDatasetParams()
Dreduce_dataset_op_test.cc32 DataTypeVector output_dtypes, in ReduceDatasetParams()
Drange_dataset_op.cc131 DataTypeVector output_dtypes) in Dataset()
144 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::RangeDatasetOp::Dataset
Diterator_ops.h69 const DataTypeVector& output_dtypes() const { return output_dtypes_; } in output_dtypes() function
Dskip_dataset_op_test.cc26 DataTypeVector output_dtypes, in SkipDatasetParams()
Dwindow_dataset.cc44 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::data::__anonb5d693710111::WindowDataset
Dzip_dataset_op_test.cc29 DataTypeVector output_dtypes, in ZipDatasetParams()
Dtensor_dataset_op.cc55 const DataTypeVector& output_dtypes() const override { return dtypes_; } in output_dtypes() function in tensorflow::data::TensorDatasetOp::Dataset
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc239 Status GetOutputDTypes(EagerOperation* op, DataTypeVector* output_dtypes) { in GetOutputDTypes()
632 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in AddOrExecuteNode() local
799 const DataTypeVector& output_dtypes, in StoreResourceDtypesAndShapes()
915 DataTypeVector output_dtypes; in EagerRemoteExecute() local
1002 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in GetKernelOutputs() local
1404 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in EagerLocalExecuteAsync() local
Dkernel_and_device.h235 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
318 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc101 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anone64ec8730111::DenseToSparseBatchDatasetOp::Dataset
102 static DataTypeVector* output_dtypes = new DataTypeVector({DT_VARIANT}); in output_dtypes() local
Dchoose_fastest_branch_dataset_op.cc40 const DataTypeVector* output_dtypes, in WrapperDataset()
48 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anond986aafb0111::WrapperDataset
230 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anond986aafb0111::ChooseFastestBranchDatasetOp::Dataset
Dauto_shard_dataset_op_test.cc29 DataTypeVector output_dtypes, in AutoShardDatasetParams()
Dstats_dataset_ops.cc68 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon8343b5b60111::LatencyStatsDatasetOp::Dataset
191 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon8343b5b60111::BytesProducedStatsDatasetOp::Dataset
Dthreadpool_dataset_op.cc163 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon3c251a260111::ThreadPoolDatasetOp::Dataset
293 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon3c251a260111::MaxIntraOpParallelismDatasetOp::Dataset
418 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon3c251a260111::PrivateThreadPoolDatasetOp::Dataset
Dio_ops.cc163 const DataTypeVector& output_dtypes, in WriteMetadataFile()
199 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::data::experimental::LoadDatasetOp::Dataset
Drandom_dataset_op_test.cc47 RandomDatasetParams(int64 seed, int64 seed2, DataTypeVector output_dtypes, in RandomDatasetParams()
Dnon_serializable_dataset_op.cc61 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon5926aa950111::NonSerializableDatasetOp::Dataset
Dignore_errors_dataset_op.cc56 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon7bd8a2940111::IgnoreErrorsDatasetOp::Dataset
Dsleep_dataset_op.cc60 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anonf7eaadcd0111::SleepDatasetOp::Dataset
Drandom_dataset_op.cc49 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::RandomDatasetOp::Dataset
Dlmdb_dataset_op.cc44 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::LMDBDatasetOp::Dataset
/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py39 def output_dtypes(self): member in CategoricalTest

1234