Searched refs:hlo_shape (Results 1 – 2 of 2) sorted by relevance
47 const Shape& hlo_shape = hlo->shape(); in IsNonComplexMatrixVectorDot() local48 return !ShapeUtil::ElementIsComplex(hlo_shape) && in IsNonComplexMatrixVectorDot()49 hlo->opcode() == HloOpcode::kDot && hlo_shape.dimensions_size() == 2 && in IsNonComplexMatrixVectorDot()50 (hlo_shape.dimensions(0) == 1 || hlo_shape.dimensions(1) == 1); in IsNonComplexMatrixVectorDot()
181 const Shape& hlo_shape = hlo.shape(); in BindHloToIrValue() local186 InsertOrDie(&base_ptrs_, &hlo, ShapeTree<llvm::Value*>(hlo_shape, nullptr)); in BindHloToIrValue()