Searched defs:broadcast_sizes (Results 1 – 7 of 7) sorted by relevance
48 Tensor::SizesType broadcast_sizes[kTensorDimensionLimit]; in masked_select_out() local
269 Tensor::SizesType broadcast_sizes[kTensorDimensionLimit]; in get_index_out_target_size() local
339 std::vector<int64_t> broadcast_sizes; in EinsumInverseDiagonal() local
342 absl::Span<const int64_t> broadcast_sizes = {}) { in Uint128AddUint64()
2976 const Shape& operand, absl::Span<const int64_t> broadcast_sizes) { in InferBroadcastShape()
933 absl::Span<const int64_t> broadcast_sizes) { in Broadcast()4054 absl::Span<const int64_t> broadcast_sizes) { in Broadcast()
490 std::vector<int64_t> broadcast_sizes; in sparse_broadcast_to() local