Searched refs:ehs (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 67 const Shape& ehs);
71 const HloInstruction* ehs);
|
| D | user_computation.cc | 1418 TF_ASSIGN_OR_RETURN(const OperationRequest* ehs, in AddTernaryInstruction()
1419 LookUpRequest(ternary_request.ehs())); in AddTernaryInstruction()
1423 rhs->output_shape(), ehs->output_shape())); in AddTernaryInstruction()
1953 PureFunctionalVisitor(session_computation, ternary_op_request.ehs(), in PureFunctionalVisitor()
2682 apply(ternary_op_request.ehs()); in ForEachOperand()
3315 HloInstruction* ehs = lookup_instruction(ternary_op_request.ehs()); in Visit() local
3330 if (!ShapeUtil::SameDimensions(request.output_shape(), ehs->shape())) { in Visit()
3331 ehs = in Visit()
3332 ImplicitBroadcastToExplicitBroadcast(ehs, request.output_shape()); in Visit()
3337 request.output_shape(), hlo_opcode, lhs, rhs, ehs)); in Visit()
|
| D | shape_inference.cc | 1027 const HloInstruction* ehs) { in InferTernaryOpShape() argument
1029 rhs->shape(), ehs->shape()); in InferTernaryOpShape()
1034 const Shape& ehs) { in InferTernaryOpShape() argument
1037 TF_DCHECK_OK(ShapeUtil::ValidateShapeWithOptionalLayout(ehs)); in InferTernaryOpShape()
1040 return InferClampShape(lhs, rhs, ehs); in InferTernaryOpShape()
1042 return InferSelectShape(lhs, rhs, ehs); in InferTernaryOpShape()
|
| D | hlo_evaluator.cc | 2002 const auto* ehs = instruction->operand(2); in ElementwiseTernaryOp() local
2008 ShapeUtil::SameDimensions(rhs->shape(), ehs->shape()))) { in ElementwiseTernaryOp()
2015 ShapeUtil::HumanString(ehs->shape()).c_str()); in ElementwiseTernaryOp()
2020 const Literal& ehs_literal = parent_->GetEvaluatedLiteralFor(ehs); in ElementwiseTernaryOp()
|
| D | hlo_instruction.h | 239 HloInstruction* ehs);
|
| D | hlo_instruction.cc | 287 HloInstruction* rhs, HloInstruction* ehs) { in CreateTernary() argument
298 return CreateNary(shape, opcode, {lhs, rhs, ehs}); in CreateTernary()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 538 const ComputationDataHandle& ehs), \
539 (lhs, rhs, ehs))
|
| D | local_computation_builder.h | 284 const ComputationDataHandle& ehs))
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | computation_builder.cc | 1107 const ComputationDataHandle& rhs, const ComputationDataHandle& ehs) { in TernaryOp() argument
1113 *request->mutable_ehs() = ehs; in TernaryOp()
|
| D | computation_builder.h | 891 const ComputationDataHandle& ehs);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 893 ComputationDataHandle ehs = 5; field
|