Searched refs:out_dim_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 783 XlaOp operand, const absl::Span<const int64> out_dim_size, in BroadcastInDim() argument
790 ShapeUtil::MakeShape(operand_shape->element_type(), out_dim_size); in BroadcastInDim()
799 broadcast_dimensions[i] > out_dim_size.size()) { in BroadcastInDim()
810 std::vector<int64> in_dim_size(out_dim_size.begin(), out_dim_size.end()); in BroadcastInDim()
2899 const absl::Span<const int64> out_dim_size, in BroadcastInDim() argument
2901 return operand.builder()->BroadcastInDim(operand, out_dim_size, in BroadcastInDim()
|
| D | xla_builder.h | 381 const absl::Span<const int64> out_dim_size,
758 XlaOp operand, const absl::Span<const int64> out_dim_size,
1204 XlaOp BroadcastInDim(XlaOp operand, const absl::Span<const int64> out_dim_size,
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 356 <b> `BroadcastInDim(operand, out_dim_size, broadcast_dimensions)` </b>
361 | `out_dim_size` | `ArraySlice<int64>` | The sizes of the dimensions |
370 The `operand` is broadcast to the shape described by `out_dim_size`.
|