Searched refs:leaf_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 89 leaf_index=None): argument
114 if leaf_index is not None:
115 result[LEAF_INDEX] = leaf_index
490 leaf_index = None
492 predictions, _, leaf_index = (
510 leaf_index = None
539 ensemble_stats, used_handlers, leaf_index)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation.h | 265 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index,
442 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index,
|
| D | hlo_computation.cc | 627 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index, in DeepCopyHelper() argument
676 HloInstruction* leaf, const ShapeIndex& leaf_index, in DeepCopyInstruction()
678 if (indices_to_copy == nullptr || indices_to_copy->element(leaf_index)) { in DeepCopyInstruction()
682 *copies_added->mutable_element(leaf_index) = copy; in DeepCopyInstruction()
696 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index, in DeepCopyInstructionWithCustomCopier() argument
|
| D | bfloat16_propagation.cc | 645 [hlo](HloInstruction* leaf, const ShapeIndex& leaf_index, in ResolveInconsistentFusions()
648 ShapeUtil::GetSubshape(hlo->shape(), leaf_index); in ResolveInconsistentFusions()
742 operand, [](HloInstruction* leaf, const ShapeIndex& leaf_index, in Run()
|