Home
last modified time | relevance | path

Searched refs:branch_operands (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Ddynamic_shaped_ops.cc169 absl::Span<const XlaOp> branch_operands) { in DynamicConditional() argument
186 branch_operands); in DynamicConditional()
215 builder->GetShape(branch_operands[i])); in DynamicConditional()
228 branch_operands); in DynamicConditional()
Ddynamic_shaped_ops.h41 absl::Span<const XlaOp> branch_operands);
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h265 absl::Span<const Shape> branch_operands);
Dshape_inference.cc2903 absl::Span<const Shape> branch_operands) { in InferConditionalShape() argument
2914 TF_RET_CHECK(branch_computations.size() == branch_operands.size()); in InferConditionalShape()
2923 branch_operands[j])) { in InferConditionalShape()
2926 ShapeUtil::HumanString(branch_operands[j]), in InferConditionalShape()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc1025 llvm::SmallVector<mlir::Value, 4> branch_operands; in matchAndRewrite() local
1026 branch_operands.push_back( in matchAndRewrite()
1029 op, adaptor.getOperands().drop_front(), &branch_operands, rewriter, in matchAndRewrite()
1055 op.getLoc(), result_types, index_value, branches, branch_operands); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h864 absl::Span<const XlaOp> branch_operands);
1463 absl::Span<const XlaOp> branch_operands);
1467 absl::Span<const XlaOp> branch_operands);
1524 absl::Span<const XlaOp> branch_operands);
2587 absl::Span<const XlaOp> branch_operands);
Dxla_builder.cc2519 absl::Span<const XlaOp> branch_operands) { in Conditional() argument
2528 return ConditionalImpl(branch_index, branch_computations, branch_operands); in Conditional()
2535 absl::Span<const XlaOp> branch_operands) { in ConditionalImpl() argument
2541 std::vector<Shape> branch_operand_shapes(branch_operands.size()); in ConditionalImpl()
2544 for (int j = 0, end = branch_operands.size(); j < end; ++j) { in ConditionalImpl()
2546 GetShape(branch_operands[j])); in ConditionalImpl()
2561 for (const XlaOp branch_operand : branch_operands) { in ConditionalImpl()
4888 absl::Span<const XlaOp> branch_operands) { in Conditional() argument
4890 branch_operands); in Conditional()
/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
Dops.pyi142 branch_operands: Sequence[XlaOp]) -> XlaOp: ...
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc980 llvm::SmallVector<xla::XlaOp, 4> branch_operands(branches.size()); in ExportXlaOp() local
1004 branch_operands[i] = args.size() == 1 ? args[0] : Tuple(ctx.builder, args); in ExportXlaOp()
1020 xla::XlaOp caseop = xla::Conditional(index, computations_p, branch_operands); in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/IR/
Dlhlo_ops.td583 `branch_operands[b]` and will be invoked with `branch_operands[b]`. The type
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md718 <b> `Conditional(branch_index, branch_computations, branch_operands)` </b>
728 | `branch_operands` | sequence of N `XlaOp` | Arguments of type \\( T_0 , |
738 will be invoked with `branch_operands[b]` which must be of the same type. The