Searched refs:out_dim_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 891 XlaOp operand, const absl::Span<const int64> out_dim_size, in BroadcastInDim() argument
899 operand_shape->element_type(), out_dim_size)); in BroadcastInDim()
908 const tensorflow::int64 num_dims = out_dim_size.size(); in BroadcastInDim()
920 std::vector<int64> in_dim_size(out_dim_size.begin(), out_dim_size.end()); in BroadcastInDim()
4103 const absl::Span<const int64> out_dim_size, in BroadcastInDim() argument
4105 return operand.builder()->BroadcastInDim(operand, out_dim_size, in BroadcastInDim()
|
| D | xla_builder.h | 456 const absl::Span<const int64> out_dim_size,
1062 XlaOp operand, const absl::Span<const int64> out_dim_size,
1651 XlaOp BroadcastInDim(XlaOp operand, const absl::Span<const int64> out_dim_size,
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 399 <b> `BroadcastInDim(operand, out_dim_size, broadcast_dimensions)` </b>
404 | `out_dim_size` | `ArraySlice<int64>` | The sizes of the dimensions |
413 The `operand` is broadcast to the shape described by `out_dim_size`.
|