Home
last modified time | relevance | path

Searched refs:ComputeMoveVector (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils_test.cc28 TEST(DatasetUtilsTest, ComputeMoveVector) { in TEST() argument
44 EXPECT_EQ(test_case.expected, ComputeMoveVector(test_case.indices)); in TEST()
Ddataset_utils.h43 std::vector<bool> ComputeMoveVector(const std::vector<int>& indices);
Dparallel_map_dataset_op.cc101 : ComputeMoveVector(indices)) { in Dataset()
Dmap_dataset_op.cc64 std::vector<bool> can_move = ComputeMoveVector(short_circuit_indices_); in MakeDataset()
Ddataset_utils.cc70 std::vector<bool> ComputeMoveVector(const std::vector<int>& indices) { in ComputeMoveVector() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc104 std::vector<bool> can_move = ComputeMoveVector(short_circuit_indices_); in MakeDataset()