Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dmap_defun.py28 output_dtypes, argument
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.h143 DataTypeVector output_dtypes() const { return output_dtypes_; } in output_dtypes() function
219 DataTypeVector output_dtypes, in BatchDatasetParams()
256 DataTypeVector type_arguments, DataTypeVector output_dtypes, in MapDatasetParams()
327 DataTypeVector output_dtypes, in TakeDatasetParams()
357 DataTypeVector output_dtypes, in ConcatenateDatasetParams()
386 DataTypeVector output_dtypes, in OptionsDatasetParams()
/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::__anon93ed33580111::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::__anon872515e00111::WrapperDataset
230 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon872515e00111::ChooseFastestBranchDatasetOp::Dataset
Dio_ops.cc197 const DataTypeVector& output_dtypes, in WriteMetadataFile()
234 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::SaveDatasetV2Op::Dataset
469 const DataTypeVector& output_dtypes, in WriteMetadataFile()
573 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::data::experimental::LoadDatasetOp::Dataset
Dstats_dataset_ops.cc68 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon1463133b0111::LatencyStatsDatasetOp::Dataset
191 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon1463133b0111::BytesProducedStatsDatasetOp::Dataset
Dauto_shard_dataset_op_test.cc29 int64_t num_replicas, DataTypeVector output_dtypes, in AutoShardDatasetParams()
Dthreadpool_dataset_op.cc174 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::ThreadPoolDatasetOp::Dataset
294 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::MaxIntraOpParallelismDatasetOp::Dataset
435 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::PrivateThreadPoolDatasetOp::Dataset
Drandom_dataset_op_test.cc48 RandomDatasetParams(int64_t seed, int64_t 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::__anon1408725a0111::NonSerializableDatasetOp::Dataset
Dsleep_dataset_op.cc60 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon4221eeb20111::SleepDatasetOp::Dataset
Dignore_errors_dataset_op.cc56 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon5ca0f8790111::IgnoreErrorsDatasetOp::Dataset
Dlmdb_dataset_op.cc44 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::LMDBDatasetOp::Dataset
Dlmdb_dataset_op_test.cc36 DataTypeVector output_dtypes, in LMDBDatasetParams()
/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op_test.cc29 DataTypeVector output_dtypes, in OptimizeDatasetParams()
Dget_options_op_test.cc35 GetOptionsParams(T input_dataset_params, DataTypeVector output_dtypes, in GetOptionsParams()
Doptions_dataset_op.cc63 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::OptionsDatasetOp::Dataset
Drange_dataset_op.cc137 DataTypeVector output_dtypes) in Dataset()
150 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::RangeDatasetOp::Dataset
Dreduce_dataset_op_test.cc32 DataTypeVector output_dtypes, in ReduceDatasetParams()
Diterator_ops.h69 const DataTypeVector& output_dtypes() const { return output_dtypes_; } in output_dtypes() function
Dzip_dataset_op_test.cc29 DataTypeVector output_dtypes, in ZipDatasetParams()
Dskip_dataset_op_test.cc26 DataTypeVector output_dtypes, in SkipDatasetParams()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h239 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
325 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
Dexecute.cc255 Status GetOutputDTypes(EagerOperation* op, DataTypeVector* output_dtypes) { in GetOutputDTypes()
1061 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in AddOrExecuteNode() local
1228 const DataTypeVector& output_dtypes, in StoreResourceDtypesAndShapes()
1342 DataTypeVector output_dtypes; in EagerRemoteExecute() local
1429 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in GetKernelOutputs() local
1841 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in EagerLocalExecuteAsync() local
/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py39 def output_dtypes(self): member in CategoricalTest

12345