Home
last modified time | relevance | path

Searched refs:output_dtypes (Results 1 – 25 of 115) sorted by relevance

12345

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op.cc73 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::DirectedInterleaveDatasetOp::Dataset
74 return data_inputs_[0]->output_dtypes(); in output_dtypes()
291 selector_input->output_dtypes().size() == 1 && in MakeDataset()
292 selector_input->output_dtypes()[0] == DT_INT64 && in MakeDataset()
305 OP_REQUIRES(ctx, data_inputs[0]->output_dtypes() == input->output_dtypes(), in MakeDataset()
309 DataTypeVectorString(data_inputs[0]->output_dtypes()), in MakeDataset()
311 DataTypeVectorString(input->output_dtypes()))); in MakeDataset()
Ddense_to_sparse_batch_dataset_op.cc35 ctx, input->output_dtypes().size() == 1, in MakeDataset()
63 switch (input->output_dtypes()[0]) { in MakeDataset()
70 input->output_dtypes()[0])); in MakeDataset()
101 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
103 return *output_dtypes; in output_dtypes()
243 DatasetIterator<Dataset<T>>::dataset()->input_->output_dtypes()[0], in GetNextInternal()
Drebatch_dataset_op.cc80 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon714a254d0111::RebatchDatasetOp::Dataset
218 input_descriptors_.resize(dataset()->output_dtypes().size()); in RestoreInternal()
337 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon714a254d0111::RebatchDatasetV2Op::Dataset
445 const size_t num_components = dataset()->output_dtypes().size(); in GetNextInternal()
454 for (int i = 0; i < dataset()->output_dtypes().size(); ++i) { in GetNextInternal()
458 out_tensors->push_back(Tensor(dataset()->output_dtypes()[i])); in GetNextInternal()
474 Tensor(dataset()->output_dtypes()[i], tensor_shape)); in GetNextInternal()
501 dataset()->output_dtypes()[i], in GetNextInternal()
556 tensors_.resize(dataset()->output_dtypes().size()); in RestoreInternal()
Dio_ops.cc56 WriteMetadataFile(ctx->env(), path, run_id, dataset->output_dtypes(), in DoCompute()
67 dataset->output_dtypes(), num_elements, in DoCompute()
122 dataset->output_dtypes(), [&mu, &status](Status s) { in WriteData()
163 const DataTypeVector& output_dtypes, in WriteMetadataFile() argument
170 for (const auto& output_dtype : output_dtypes) { in WriteMetadataFile()
199 const DataTypeVector& output_dtypes() const override { return output_types_; } in output_dtypes() function in tensorflow::data::experimental::LoadDatasetOp::Dataset
306 dataset()->metadata_.version(), dataset()->output_dtypes(), in InitializeInput()
Dunique_dataset_op.cc45 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::UniqueDatasetOp::Dataset
46 return input_->output_dtypes(); in output_dtypes()
215 OP_REQUIRES(ctx, input->output_dtypes().size() == 1, in MakeDataset()
219 DataType input_dtype = input->output_dtypes()[0]; in MakeDataset()
Dstats_dataset_ops.cc68 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon8343b5b60111::LatencyStatsDatasetOp::Dataset
69 return input_->output_dtypes(); in output_dtypes()
191 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon8343b5b60111::BytesProducedStatsDatasetOp::Dataset
192 return input_->output_dtypes(); in output_dtypes()
Dthreadpool_dataset_op.cc163 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon3c251a260111::ThreadPoolDatasetOp::Dataset
164 return input_->output_dtypes(); in output_dtypes()
293 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon3c251a260111::MaxIntraOpParallelismDatasetOp::Dataset
294 return input_->output_dtypes(); in output_dtypes()
418 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon3c251a260111::PrivateThreadPoolDatasetOp::Dataset
419 return input_->output_dtypes(); in output_dtypes()
Dauto_shard_dataset_op_test.cc29 DataTypeVector output_dtypes, in AutoShardDatasetParams() argument
32 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in AutoShardDatasetParams()
Dscan_dataset_op.cc100 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp::Dataset
201 output_dtypes().size()); in GetNextInternal()
206 dataset()->state_types_.size() + output_dtypes().size()); in GetNextInternal()
221 if (state_and_output[i].dtype() != output_dtypes()[output_index]) { in GetNextInternal()
Dunbatch_dataset_op.cc67 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::experimental::__anon91d1b6390111::UnbatchDatasetOp::Dataset
68 return input_->output_dtypes(); in output_dtypes()
223 tensors_.resize(dataset()->output_dtypes().size()); in RestoreInternal()
Dsampling_dataset_op_test.cc29 DataTypeVector output_dtypes, in SamplingDatasetParams() argument
32 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in SamplingDatasetParams()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dmap_defun.py28 output_dtypes, argument
55 if not isinstance(output_dtypes, list):
67 output_dtypes, output_shapes, concrete_fn,
/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op.cc64 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::ConcatenateDatasetOp::Dataset
65 return input_->output_dtypes(); in output_dtypes()
223 OP_REQUIRES(ctx, input->output_dtypes() == to_concatenate->output_dtypes(), in MakeDataset()
227 (DataTypeVectorString(input->output_dtypes()), in MakeDataset()
228 DataTypeVectorString(to_concatenate->output_dtypes())))); in MakeDataset()
Dconcatenate_dataset_op_test.cc131 SameShapeConcatenateDatasetParams().output_dtypes()}, in DatasetOutputDtypesTestCases()
134 DifferentShapeConcatenateDatasetParams().output_dtypes()}}; in DatasetOutputDtypesTestCases()
169 SameShapeConcatenateDatasetParams().output_dtypes()}, in DATASET_CARDINALITY_TEST_P()
172 DifferentShapeConcatenateDatasetParams().output_dtypes()}}; in DATASET_CARDINALITY_TEST_P()
Dflat_map_dataset_op_test.cc30 DataTypeVector output_dtypes, in FlatMapDatasetParams() argument
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in FlatMapDatasetParams()
149 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
167 TF_ASSERT_OK(CheckIteratorOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
Drange_dataset_op.cc131 DataTypeVector output_dtypes) in Dataset() argument
136 output_dtypes_(output_dtypes) {} in Dataset()
144 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::RangeDatasetOp::Dataset
229 switch (dataset()->output_dtypes()[0]) { in GetNextInternal()
240 DataTypeString(dataset()->output_dtypes()[0])); in GetNextInternal()
Ddataset_test_base.h123 DatasetParams(DataTypeVector output_dtypes,
143 DataTypeVector output_dtypes() const { return output_dtypes_; } in output_dtypes() function
180 DataTypeVector output_dtypes,
187 DataTypeVector output_dtypes);
210 DataTypeVector output_dtypes, in BatchDatasetParams() argument
213 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in BatchDatasetParams()
247 DataTypeVector type_arguments, DataTypeVector output_dtypes, in MapDatasetParams() argument
251 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in MapDatasetParams()
318 DataTypeVector output_dtypes, in TakeDatasetParams() argument
321 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in TakeDatasetParams()
[all …]
Doptimize_dataset_op_test.cc29 DataTypeVector output_dtypes, in OptimizeDatasetParams() argument
33 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in OptimizeDatasetParams()
Dpadded_batch_dataset_op.cc99 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::PaddedBatchDatasetOp::Dataset
100 return input_->output_dtypes(); in output_dtypes()
167 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal()
292 output_dtypes()[component_index], in GetNextInternal()
447 OP_REQUIRES(ctx, padding_value_t.dtype() == input->output_dtypes()[i], in MakeDataset()
452 DataTypeString(input->output_dtypes()[i]))); in MakeDataset()
Dtensor_slice_dataset_op_test.cc155 PlainTensorSliceDatasetParams().output_dtypes()}, in DatasetOutputTypesTestCases()
157 NestedTensorSliceDatasetParams().output_dtypes()}}; in DatasetOutputTypesTestCases()
186 PlainTensorSliceDatasetParams().output_dtypes()}, in DATASET_CARDINALITY_TEST_P()
188 NestedTensorSliceDatasetParams().output_dtypes()}}; in DATASET_CARDINALITY_TEST_P()
Dzip_dataset_op.cc42 for (DataType dt : input->output_dtypes()) { in Dataset()
63 const DataTypeVector& output_dtypes() const override { in output_dtypes() function in tensorflow::data::ZipDatasetOp::Dataset
146 out_tensors->reserve(dataset()->output_dtypes().size()); in GetNextInternal()
Dwindow_dataset_op_test.cc28 DataTypeVector output_dtypes, in WindowDatasetParams() argument
31 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in WindowDatasetParams()
346 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F()
385 TF_ASSERT_OK(CheckIteratorOutputDtypes(dataset_params.output_dtypes())); in DATASET_CARDINALITY_TEST_P()
Dreduce_dataset_op_test.cc32 DataTypeVector output_dtypes, in ReduceDatasetParams() argument
35 : DatasetParams(std::move(output_dtypes), std::move(output_shapes), in ReduceDatasetParams()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc239 Status GetOutputDTypes(EagerOperation* op, DataTypeVector* output_dtypes) { in GetOutputDTypes() argument
251 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, output_dtypes)); in GetOutputDTypes()
632 const DataTypeVector& output_dtypes = kernel->output_dtypes(); in AddOrExecuteNode() local
639 output_dtypes[i], &ctx); in AddOrExecuteNode()
642 CreateUnshapedOutput(*kernel, i, output_device, output_dtypes[i], in AddOrExecuteNode()
799 const DataTypeVector& output_dtypes, in StoreResourceDtypesAndShapes() argument
802 if (output_dtypes.size() != 1) { in StoreResourceDtypesAndShapes()
805 if (output_dtypes[0] != DT_RESOURCE) { in StoreResourceDtypesAndShapes()
915 DataTypeVector output_dtypes; in EagerRemoteExecute() local
916 TF_RETURN_IF_ERROR(GetOutputDTypes(op, &output_dtypes)); in EagerRemoteExecute()
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dcategorical_op_test.py39 def output_dtypes(self): member in CategoricalTest
105 for output_dtype in self.output_dtypes():
110 for output_dtype in self.output_dtypes():
145 for output_dtype in self.output_dtypes():

12345