| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 52 ShapeUtil::ForEachSubshape( in DetermineFusionComputationPrecision()
100 ShapeUtil::ForEachSubshape( in RevertIfFusionInternalBF16Changes()
176 ShapeUtil::ForEachSubshape( in DetermineWhileComputationsPrecision()
341 ShapeUtil::ForEachSubshape(hlo->shape(), [&](const Shape& /* subshape */, in DetermineInstructionPrecision()
376 ShapeUtil::ForEachSubshape( in DetermineInstructionPrecision()
415 ShapeUtil::ForEachSubshape( in AdjustCalledComputationParameters()
456 ShapeUtil::ForEachSubshape(root->shape(), [this, hlo, root, output]( in AdjustCalledComputationRoot()
552 ShapeUtil::ForEachSubshape(hlo->shape(), adjust_hlo_output); in ResolveInconsistencyOfAliasingBuffersHelper()
569 ShapeUtil::ForEachSubshape(hlo->shape(), adjust_hlo_output); in ResolveInconsistencyOfAliasingBuffersHelper()
|
| D | while_loop_invariant_code_motion.cc | 221 ShapeUtil::ForEachSubshape( in TryHoistingInvariantInstructionsFromWhileBody()
229 ShapeUtil::ForEachSubshape( in TryHoistingInvariantInstructionsFromWhileBody()
|
| D | allocation_tracker.cc | 79 ShapeUtil::ForEachSubshape( in RegisterInternal()
112 ShapeUtil::ForEachSubshape( in Unregister()
|
| D | optimize_input_output_buffer_alias.cc | 51 ShapeUtil::ForEachSubshape( in Build()
|
| D | logical_buffer_analysis.cc | 102 ShapeUtil::ForEachSubshape( in DefaultAction()
|
| D | shaped_buffer.cc | 84 ShapeUtil::ForEachSubshape( in ToString()
|
| D | instruction_fusion.cc | 174 ShapeUtil::ForEachSubshape( in EffectivelyAtMostUnary()
273 ShapeUtil::ForEachSubshape( in ComputeGloballyUnfusible()
|
| D | buffer_assignment.cc | 1563 ShapeUtil::ForEachSubshape( in BuildColocatedBufferSets()
1596 ShapeUtil::ForEachSubshape( in BuildColocatedBufferSets()
1608 ShapeUtil::ForEachSubshape( in BuildColocatedBufferSets()
1624 ShapeUtil::ForEachSubshape( in BuildColocatedBufferSets()
1646 ShapeUtil::ForEachSubshape( in BuildColocatedBufferSets()
|
| D | hlo_cost_analysis.cc | 583 ShapeUtil::ForEachSubshape(crs->shape(), in HandleAllReduce()
623 ShapeUtil::ForEachSubshape( in HandleFusion()
|
| D | hlo_liveness_analysis.cc | 97 ShapeUtil::ForEachSubshape( in MarkLiveAtAllIndices()
|
| D | while_loop_simplifier.cc | 602 ShapeUtil::ForEachSubshape( in UnflattenTupleInstr()
664 ShapeUtil::ForEachSubshape(while_shape, in TryFlattenNestedTuples()
|
| D | hlo_alias_analysis.cc | 459 ShapeUtil::ForEachSubshape( in ToString()
|
| D | hlo_rematerialization.cc | 1222 ShapeUtil::ForEachSubshape( in Run()
|
| D | copy_insertion.cc | 1038 ShapeUtil::ForEachSubshape( in AddSpecialCaseCopies()
|
| D | hlo_graph_dumper.cc | 1138 ShapeUtil::ForEachSubshape(instr->shape(), in GetInstructionNodeExtraInfo()
|
| D | buffer_assignment_test.cc | 900 ShapeUtil::ForEachSubshape( in TEST_F()
1286 ShapeUtil::ForEachSubshape( in TEST_F()
|
| D | hlo_verifier.cc | 1055 ShapeUtil::ForEachSubshape( in ShapeContainsToken()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | alias_analysis.cc | 143 ShapeUtil::ForEachSubshape( in GetNoaliasMetadataForBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 96 ShapeUtil::ForEachSubshape( in EmitBasePointersForHlos()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 353 ForEachSubshape(shape, [&](const Shape& literal_subshape, in SubshapeCount()
836 ForEachSubshape(shape, [&](const Shape& sub_shape, const ShapeIndex& index) { in GetLeafShapes()
892 /* static */ void ShapeUtil::ForEachSubshape(const Shape& shape, in ForEachSubshape() function in xla::ShapeUtil
|
| D | shape_util_test.cc | 478 ShapeUtil::ForEachSubshape( in TEST()
493 ShapeUtil::ForEachSubshape( in TEST()
|
| D | literal_util.cc | 62 ShapeUtil::ForEachSubshape( in ConvertType()
|
| D | shape_util.h | 513 static void ForEachSubshape(const Shape& shape, const VisitorFunction& func);
|
| D | literal.cc | 596 ShapeUtil::ForEachSubshape( in Relayout()
921 ShapeUtil::ForEachSubshape( in Hash()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 190 ShapeUtil::ForEachSubshape( in DestructureTuple()
|