Searched refs:slices_a (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
D | choose_fastest_dataset_test.py | 70 slices_a=[[0]], 74 slices_a=[[0]], 78 slices_a=[[0]], 82 slices_a=[[1, 2, 3]], 85 def testChooseFastestErrorWithIncompatibleInput(self, slices_a, slices_b, argument 87 dataset_a = dataset_ops.Dataset.from_tensor_slices(slices_a)
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | buffer_assignment.cc | 539 SliceSet slices_a = collect_slices(hlo_a); in HaveDisjointSlices() local 544 return !slices_a.empty() && !slices_b.empty() && in HaveDisjointSlices() 545 absl::c_none_of(slices_a, [&](const BufferAllocation::Slice& slice) { in HaveDisjointSlices()
|