Searched defs:to_apply (Results 1 – 5 of 5) sorted by relevance
64 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape() local317 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local567 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F() local638 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F() local716 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local728 ProgramShape to_apply = in TEST_F() local741 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local753 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local764 ProgramShape to_apply = in TEST_F() local777 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local[all …]
1073 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()2045 const ProgramShape& to_apply) { in InferReduceShape()2834 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
819 optional<HloComputation*> to_apply; in ParseInstructionRhs() local923 optional<HloComputation*> to_apply; in ParseInstructionRhs() local1036 optional<HloComputation*> to_apply; in ParseInstructionRhs() local1180 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
426 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters() local
1988 HloComputation* HloInstruction::to_apply() const { in to_apply() function in xla::HloInstruction