Home
last modified time | relevance | path

Searched defs:to_apply (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc66 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape() local
329 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
944 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F() local
1015 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F() local
1093 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1107 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1131 ProgramShape to_apply = in TEST_F() local
1144 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1156 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1169 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
[all …]
Dhlo_parser.cc1227 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
1399 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
1547 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
1786 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
2244 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
Dshape_inference.cc1117 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2116 const ProgramShape& to_apply) { in InferReduceShape()
3179 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
Dhlo_verifier.cc644 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters() local
Dhlo_instruction.cc1581 HloComputation* to_apply, absl::string_view custom_call_target, in CreateCustomCall()
2428 HloComputation* HloInstruction::to_apply() const { in to_apply() function in xla::HloInstruction
Dhlo_instructions.cc2356 HloComputation* to_apply, absl::string_view custom_call_target, in HloCustomCallInstruction()