Searched defs:to_apply (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 66 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape() local
329 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
965 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F() local
1036 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, f32_); in TEST_F() local
1114 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1128 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1152 ProgramShape to_apply = in TEST_F() local
1165 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1177 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
1190 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F() local
[all …]
|
| D | shape_inference.cc | 1106 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2230 const ProgramShape& to_apply) { in InferReduceShape()
3316 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
|
| D | hlo_parser.cc | 1298 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
1522 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
1684 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
1923 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
2381 optional<HloComputation*> to_apply; in ParseInstructionRhs() local
|
| D | hlo_verifier.cc | 981 const ProgramShape& to_apply = instruction.to_apply()->ComputeProgramShape(); in SameElementTypesForOperandsAndToApplyParameters() local
|
| D | hlo_instruction.cc | 1765 HloComputation* to_apply, absl::string_view custom_call_target, in CreateCustomCall()
2625 HloComputation* HloInstruction::to_apply() const { in to_apply() function in xla::HloInstruction
|
| D | hlo_instructions.cc | 2497 HloComputation* to_apply, absl::string_view custom_call_target, in HloCustomCallInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 410 XlaComputation to_apply; in TEST_F() local
|