Home
last modified time | relevance | path

Searched defs:hlo_shape (Results 1 – 3 of 3) 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.cc159 const Shape& hlo_shape = hlo.shape(); in BindHloToIrValue() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc428 auto hlo_shape = hlo->shape(); in TileToPartialReplicateHaloExchange() local
1263 const Shape& hlo_shape = hlo->shape(); in SliceFirstK() local