Searched refs:kToutputTypes (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | tensor_slice_dataset_op.cc | 33 /* static */ constexpr const char* const TensorSliceDatasetOp::kToutputTypes; member in tensorflow::data::TensorSliceDatasetOp 106 {{kToutputTypes, dtypes}}, output)); in AsGraphDefInternal() 177 OP_REQUIRES_OK(ctx, ctx->GetAttr(kToutputTypes, &output_types_)); in TensorSliceDatasetOp()
|
D | tensor_slice_dataset_op.h | 27 static constexpr const char* const kToutputTypes = "Toutput_types"; variable
|
D | padded_batch_dataset_op.h | 32 static constexpr const char* const kToutputTypes = "Toutput_types"; variable
|
D | padded_batch_dataset_op.cc | 43 /* static */ constexpr const char* const PaddedBatchDatasetOp::kToutputTypes; member in tensorflow::data::PaddedBatchDatasetOp 176 {kToutputTypes, output_types}, in AsGraphDefInternal()
|
D | padded_batch_dataset_op_test.cc | 85 {PaddedBatchDatasetOp::kToutputTypes, output_dtypes_}, in GetAttributes()
|
D | dataset_test_base.cc | 1089 *attr_vector = {{TensorSliceDatasetOp::kToutputTypes, output_dtypes_}, in GetAttributes()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | parallel_interleave_dataset_op_test.cc | 119 /*attrs=*/{{TensorSliceDatasetOp::kToutputTypes, output_types}, in MakeTensorSliceDatasetFunc()
|