Home
last modified time | relevance | path

Searched defs:hlo_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc46 const Shape& hlo_shape = hlo->shape(); in IsNonComplexNonBatchedMatrixVectorDot() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc160 const Shape& hlo_shape = hlo.shape(); in BindHloToIrValue() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc343 auto hlo_shape = hlo->shape(); in TileToPartialReplicateHaloExchange() local
1225 const Shape& hlo_shape = hlo->shape(); in SliceFirstK() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc3863 xla::Shape hlo_shape = hlo->shape(); in Postprocess() local