Home
last modified time | relevance | path

Searched defs:broadcast_sizes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc339 std::vector<int64> broadcast_sizes; in EinsumInverseDiagonal() local
Dprng.cc342 absl::Span<const int64> broadcast_sizes = {}) { in Uint128AddUint64()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2973 const Shape& operand, absl::Span<const int64> broadcast_sizes) { in InferBroadcastShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc877 absl::Span<const int64> broadcast_sizes) { in Broadcast()
3910 XlaOp Broadcast(const XlaOp operand, absl::Span<const int64> broadcast_sizes) { in Broadcast()