Searched refs:dataset_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op_test.cc | 38 Tensor dataset_tensor(DT_VARIANT, TensorShape({})); in CreateTensorSliceDatasetTensors() local
40 StoreDatasetInVariantTensor(tensor_slice_dataset, &dataset_tensor)); in CreateTensorSliceDatasetTensors()
41 dataset_tensors->emplace_back(std::move(dataset_tensor)); in CreateTensorSliceDatasetTensors()
|
| D | zip_dataset_op_test.cc | 42 Tensor dataset_tensor(DT_VARIANT, TensorShape({})); in CreateRangeDatasetTensors() local
44 StoreDatasetInVariantTensor(range_dataset, &dataset_tensor)); in CreateRangeDatasetTensors()
45 dataset_tensors->emplace_back(std::move(dataset_tensor)); in CreateRangeDatasetTensors()
|
| D | map_dataset_op_test.cc | 71 Tensor dataset_tensor(DT_VARIANT, TensorShape({})); in CreateMapDatasetContext() local
73 StoreDatasetInVariantTensor(input_dataset, &dataset_tensor)); in CreateMapDatasetContext()
74 Variant variant = dataset_tensor.scalar<Variant>()(); in CreateMapDatasetContext()
|
| D | take_dataset_op_test.cc | 27 std::vector<Tensor> *const tensor_vector, Tensor *dataset_tensor) { in CreateTensorSliceDatasetTensor() argument
32 StoreDatasetInVariantTensor(tensor_slice_dataset, dataset_tensor)); in CreateTensorSliceDatasetTensor()
|
| D | repeat_dataset_op_test.cc | 30 std::vector<Tensor> *const tensor_vector, Tensor *dataset_tensor) { in CreateTensorSliceDatasetTensor() argument
35 StoreDatasetInVariantTensor(tensor_slice_dataset, dataset_tensor)); in CreateTensorSliceDatasetTensor()
|