HomeSort by relevance Sort by last modified time
    Searched refs:indices_to_copy (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
copy_insertion.cc 91 // 'indices_to_copy'. Add control edges from the respective kCopy instructions
130 const ShapeTree<bool>& indices_to_copy) {
133 // copies. Elements which are not copied (indices_to_copy.element(index) ==
139 from, &indices_to_copy, &from_copy_tree));
144 to->parent()->DeepCopyInstruction(to, &indices_to_copy, &to_copy_tree));
170 ShapeTree<bool>* indices_to_copy) {
171 DCHECK(ShapeUtil::Compatible(indices_to_copy->shape(), xla_while->shape()));
175 for (auto& pair : *indices_to_copy) {
261 ShapeTree<bool> indices_to_copy(xla_while->shape());
263 &indices_to_copy)) {
1074 const ShapeTree<bool>& indices_to_copy = pair.second; local
    [all...]
hlo_computation_test.cc 305 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/true);
306 EXPECT_THAT(computation->DeepCopyInstruction(constant, &indices_to_copy)
313 ShapeTree<bool> indices_to_copy(constant->shape(), /*init_value=*/false);
314 EXPECT_EQ(computation->DeepCopyInstruction(constant, &indices_to_copy)
334 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/true);
338 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added)
351 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false);
355 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added)
368 ShapeTree<bool> indices_to_copy(tuple->shape(), /*init_value=*/false);
369 *indices_to_copy.mutable_element({0}) = true
    [all...]
hlo_computation.cc 658 HloInstruction* instruction, const ShapeTree<bool>* indices_to_copy,
665 if (indices_to_copy != nullptr &&
666 !ShapeUtil::Compatible(instruction->shape(), indices_to_copy->shape())) {
671 ShapeUtil::HumanString(indices_to_copy->shape()));
675 auto copy_leaf = [indices_to_copy, copies_added](
678 if (indices_to_copy == nullptr || indices_to_copy->element(leaf_index)) {
    [all...]
hlo_computation.h 250 // indices_to_copy is non-null then this ShapeTree indicates which elements
257 const ShapeTree<bool>* indices_to_copy = nullptr,

Completed in 130 milliseconds