Searched refs:CopySubtreeFrom (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree_test.cc | 258 destination.CopySubtreeFrom(source, /*source_base_index=*/{}, in TEST_F()
274 destination.CopySubtreeFrom(source, /*source_base_index=*/{}, in TEST_F()
293 destination.CopySubtreeFrom(source, /*source_base_index=*/{0}, in TEST_F()
312 destination.CopySubtreeFrom(source, /*source_base_index=*/{}, in TEST_F()
337 destination.CopySubtreeFrom(source, /*source_base_index=*/{1}, in TEST_F()
|
| D | shape_tree.h | 290 void CopySubtreeFrom(const ShapeTree<T>& other,
685 void ShapeTree<T>::CopySubtreeFrom(const ShapeTree<T>& other, in CopySubtreeFrom() function
717 sub_shape_tree.CopySubtreeFrom(*this, index, {}); in SubShapeTree()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_replication_analysis.cc | 289 shape_tree.CopySubtreeFrom(hlo_replication_[inst->operand(i)], {}, {i}); in ComputeHloReplicationOnComputation()
294 shape_tree.CopySubtreeFrom(hlo_replication_[inst->operand(0)], in ComputeHloReplicationOnComputation()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 377 (*allocation)->buffers_.CopySubtreeFrom(parent->buffers_, subshape, {}); in MakeSubBuffer()
519 allocation_tmp->buffers_.CopySubtreeFrom(element.allocation->buffers_, {}, in MakeTuple()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 158 element_shape_tree.CopySubtreeFrom(tuple_shape_tree, src_index, {}); in GetSubtree()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 446 buffer.buffers().CopySubtreeFrom(nontuple_buffer.buffers(), in PopulateOutputs()
|