Home
last modified time | relevance | path

Searched refs:broadcast_sizes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h219 const Shape& operand, absl::Span<const int64> broadcast_sizes);
Dshape_inference.cc2608 const Shape& operand, absl::Span<const int64> broadcast_sizes) { in InferBroadcastShape() argument
2610 for (int64 size : broadcast_sizes) { in InferBroadcastShape()
2618 broadcast_sizes.size()); in InferBroadcastShape()
2619 std::copy(broadcast_sizes.begin(), broadcast_sizes.end(), dimensions.begin()); in InferBroadcastShape()
2621 dimensions.begin() + broadcast_sizes.size()); in InferBroadcastShape()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc471 absl::Span<const int64> broadcast_sizes) { in Broadcast() argument
472 return xla::Broadcast(operand.op(), broadcast_sizes); in Broadcast()
Dlocal_computation_builder.h241 absl::Span<const int64> broadcast_sizes);
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h330 absl::Span<const int64> broadcast_sizes);
716 absl::Span<const int64> broadcast_sizes);
1120 XlaOp Broadcast(const XlaOp& operand, absl::Span<const int64> broadcast_sizes);
Dxla_builder.cc648 absl::Span<const int64> broadcast_sizes) { in Broadcast() argument
653 ShapeInference::InferBroadcastShape(operand_shape, broadcast_sizes)); in Broadcast()
2839 XlaOp Broadcast(const XlaOp& operand, absl::Span<const int64> broadcast_sizes) { in Broadcast() argument
2840 return operand.builder()->Broadcast(operand, broadcast_sizes); in Broadcast()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md285 <b> `Broadcast(operand, broadcast_sizes)` </b>
290 `broadcast_sizes` | `ArraySlice<int64>` | The sizes of the new dimensions
292 The new dimensions are inserted on the left, i.e. if `broadcast_sizes` has
303 `broadcast_sizes` is `{2, 3}`, then the result will be an array with shape