Searched defs:to_apply (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference_test.cc | 68 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape() local
331 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
968 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F() local
1039 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F() local
1072 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, s32_}, s32_); in TEST_F() local
1128 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1142 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1166 ProgramShape to_apply = in TEST_F() local
1179 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1191 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
[all …]
|
| H A D | shape_inference.cc | 1115 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2219 const ProgramShape& to_apply) { in InferReduceShape()
3298 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
|
| H A D | hlo_parser.cc | 1466 optional<HloComputation*> to_apply; in CreateInstruction() local
1773 optional<HloComputation*> to_apply; in CreateInstruction() local
1927 optional<HloComputation*> to_apply; in CreateInstruction() local
2157 optional<HloComputation*> to_apply; in CreateInstruction() local
2590 optional<HloComputation*> to_apply; in CreateInstruction() local
|
| H A D | hlo_instruction.cc | 1888 HloComputation* to_apply, absl::string_view custom_call_target, in CreateCustomCall()
2786 HloComputation* HloInstruction::to_apply() const { in to_apply() function in xla::HloInstruction
|
| H A D | hlo_verifier.cc | 1007 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters() local
|
| H A D | hlo_evaluator.cc | 281 HloComputation* to_apply = call_instruction->to_apply(); in PatternMatchLoopCondComparison() local
|
| H A D | hlo_instructions.cc | 2657 HloComputation* to_apply, absl::string_view custom_call_target, in HloCustomCallInstruction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder_test.cc | 412 XlaComputation to_apply; in TEST_F() local
|