Searched refs:dataset_tensors (Results 1 – 2 of 2) sorted by relevance
31 std::vector<Tensor> *const dataset_tensors) { in CreateTensorSliceDatasetTensors() argument41 dataset_tensors->emplace_back(std::move(dataset_tensor)); in CreateTensorSliceDatasetTensors()
36 std::vector<Tensor> *const dataset_tensors) { in CreateRangeDatasetTensors() argument45 dataset_tensors->emplace_back(std::move(dataset_tensor)); in CreateRangeDatasetTensors()