Searched refs:TensorSliceDatasetOp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_slice_dataset_op.cc | 31 /* static */ constexpr const char* const TensorSliceDatasetOp::kDatasetType;
32 /* static */ constexpr const char* const TensorSliceDatasetOp::kComponents;
33 /* static */ constexpr const char* const TensorSliceDatasetOp::kToutputTypes;
34 /* static */ constexpr const char* const TensorSliceDatasetOp::kOutputShapes;
38 class TensorSliceDatasetOp::Dataset : public DatasetBase {
175 TensorSliceDatasetOp::TensorSliceDatasetOp(OpKernelConstruction* ctx) in TensorSliceDatasetOp() function in tensorflow::data::TensorSliceDatasetOp
181 void TensorSliceDatasetOp::MakeDataset(OpKernelContext* ctx, in MakeDataset()
210 TensorSliceDatasetOp);
|
| D | tensor_slice_dataset_op.h | 23 class TensorSliceDatasetOp : public DatasetOpKernel {
30 explicit TensorSliceDatasetOp(OpKernelConstruction* ctx);
|
| D | tensor_slice_dataset_op_test.cc | 149 name_utils::OpName(TensorSliceDatasetOp::kDatasetType))); in TEST_F()
211 TensorSliceDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
|
| D | dataset_test_base.cc | 1082 absl::StrCat(TensorSliceDatasetOp::kComponents, "_", i)); in GetInputNames()
1089 *attr_vector = {{TensorSliceDatasetOp::kToutputTypes, output_dtypes_}, in GetAttributes()
1090 {TensorSliceDatasetOp::kOutputShapes, output_shapes_}}; in GetAttributes()
1117 return TensorSliceDatasetOp::kDatasetType; in dataset_type()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parallel_interleave_dataset_op_test.cc | 119 /*attrs=*/{{TensorSliceDatasetOp::kToutputTypes, output_types}, in MakeTensorSliceDatasetFunc()
120 {TensorSliceDatasetOp::kOutputShapes, output_shapes}}); in MakeTensorSliceDatasetFunc()
|