Searched refs:hlo_shape (Results 1 – 3 of 3) sorted by relevance
46 const Shape& hlo_shape = hlo->shape(); in IsNonComplexNonBatchedMatrixVectorDot() local47 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexNonBatchedMatrixVectorDot()48 hlo->opcode() == HloOpcode::kDot && hlo_shape.dimensions_size() <= 1 && in IsNonComplexNonBatchedMatrixVectorDot()
159 const Shape& hlo_shape = hlo.shape(); in BindHloToIrValue() local164 InsertOrDie(&base_ptrs_, &hlo, ShapeTree<llvm::Value*>(hlo_shape, nullptr)); in BindHloToIrValue()
428 auto hlo_shape = hlo->shape(); in TileToPartialReplicateHaloExchange() local1263 const Shape& hlo_shape = hlo->shape(); in SliceFirstK() local1264 auto hlo_dims = hlo_shape.dimensions(); in SliceFirstK()1265 std::vector<int64> start_indices(hlo_shape.dimensions_size(), 0); in SliceFirstK()1267 std::vector<int64> strides(hlo_shape.dimensions_size(), 1); in SliceFirstK()1269 auto output_shape = hlo_shape; in SliceFirstK()