| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | map_defun.py | 24 output_dtypes, argument
|
| /external/pytorch/test/distributed/flight_recorder/ |
| D | test_fr_analysis.py | 28 output_dtypes="float32", argument
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.h | 151 DataTypeVector output_dtypes() const { return output_dtypes_; } in output_dtypes() function
227 DataTypeVector output_dtypes, in BatchDatasetParams()
264 DataTypeVector type_arguments, DataTypeVector output_dtypes, in MapDatasetParams()
337 DataTypeVector output_dtypes, in TakeDatasetParams()
367 DataTypeVector output_dtypes, in ConcatenateDatasetParams()
396 DataTypeVector output_dtypes, in OptionsDatasetParams()
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tfrt_graph_execution_state_test.cc | 355 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
422 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
475 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
557 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
598 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
665 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
931 std::vector<tensorflow::DataType> output_dtypes = { in TEST_F() local
1040 std::vector<DataType> output_dtypes = {DT_FLOAT, DT_INT32, DT_FLOAT, in MakeOuterGraph() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | range_dataset_op.cc | 49 Status ConvertOutputTypes(const tensorflow::DataTypeVector& output_dtypes, in ConvertOutputTypes()
157 DataTypeVector output_dtypes, bool replicate_on_split) in Dataset()
171 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::RangeDatasetOp::Dataset
|
| D | rewrite_dataset_op_test.cc | 29 DataTypeVector output_dtypes, in RewriteDatasetParams()
|
| D | optimize_dataset_op_test.cc | 29 DataTypeVector output_dtypes, in OptimizeDatasetParams()
|
| D | get_options_op_test.cc | 35 GetOptionsParams(T input_dataset_params, DataTypeVector output_dtypes, in GetOptionsParams()
|
| D | reduce_dataset_op_test.cc | 32 DataTypeVector output_dtypes, in ReduceDatasetParams()
|
| D | options_dataset_op.cc | 63 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::OptionsDatasetOp::Dataset
|
| D | iterator_ops.h | 74 const DataTypeVector& output_dtypes() const { return output_dtypes_; } in output_dtypes() function
|
| D | zip_dataset_op_test.cc | 29 DataTypeVector output_dtypes, in ZipDatasetParams()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 101 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anondd8797220111::DenseToSparseBatchDatasetOp::Dataset
102 static DataTypeVector* output_dtypes = new DataTypeVector({DT_VARIANT}); in output_dtypes() local
|
| D | choose_fastest_branch_dataset_op.cc | 40 const DataTypeVector* output_dtypes, in WrapperDataset()
48 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anond0bf79aa0111::WrapperDataset
230 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anond0bf79aa0111::ChooseFastestBranchDatasetOp::Dataset
|
| D | io_ops.cc | 199 const DataTypeVector& output_dtypes, in WriteMetadataFile()
236 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::SaveDatasetV2Op::Dataset
471 const DataTypeVector& output_dtypes, in WriteMetadataFile()
575 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::data::experimental::LoadDatasetOp::Dataset
|
| D | stats_dataset_ops.cc | 68 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon87d51c450111::LatencyStatsDatasetOp::Dataset
193 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon87d51c450111::BytesProducedStatsDatasetOp::Dataset
|
| D | threadpool_dataset_op.cc | 188 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::ThreadPoolDatasetOp::Dataset
310 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::MaxIntraOpParallelismDatasetOp::Dataset
451 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::PrivateThreadPoolDatasetOp::Dataset
|
| D | random_dataset_op_test.cc | 48 RandomDatasetParams(int64_t seed, int64_t seed2, DataTypeVector output_dtypes, in RandomDatasetParams()
|
| D | non_serializable_dataset_op.cc | 61 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon75f8cfe40111::NonSerializableDatasetOp::Dataset
|
| D | ignore_errors_dataset_op.cc | 56 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anonc6e132430111::IgnoreErrorsDatasetOp::Dataset
|
| D | sleep_dataset_op.cc | 60 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon0791fe7c0111::SleepDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 36 def output_dtypes(self): member in CategoricalTest
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 244 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
340 const DataTypeVector& output_dtypes() const override { in output_dtypes() function
|
| D | execute.cc | 267 Status GetOutputDTypes(EagerOperation* op, DataTypeVector* output_dtypes) { in GetOutputDTypes()
1331 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in AddOrExecuteNode() local
1508 const DataTypeVector& output_dtypes, in StoreResourceDtypesAndShapes()
1623 DataTypeVector output_dtypes; in EagerRemoteExecute() local
1713 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in GetKernelOutputs() local
2136 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in EagerLocalExecuteAsync() local
|
| /external/federated-compute/fcp/tensorflow/ |
| D | external_dataset_op.cc | 108 const tensorflow::DataTypeVector& output_dtypes() const override { in output_dtypes() function in fcp::ExternalDatasetOp::Dataset
|