Searched refs:parameter_replicated_at_leaf_buffers (Results 1 – 3 of 3) sorted by relevance
871 absl::Span<const bool> parameter_replicated_at_leaf_buffers) { in set_parameter_replicated_at_leaf_buffers() argument873 parameter_replicated_at_leaf_buffers.size()); in set_parameter_replicated_at_leaf_buffers()875 parameter_replicated_at_leaf_buffers.begin(), in set_parameter_replicated_at_leaf_buffers()876 parameter_replicated_at_leaf_buffers.end()); in set_parameter_replicated_at_leaf_buffers()879 parameter_replicated_at_leaf_buffers() const { in parameter_replicated_at_leaf_buffers() function
1494 absl::Span<const bool> parameter_replicated_at_leaf_buffers);1498 parameter_replicated_at_leaf_buffers() const;
3372 absl::Span<const bool> parameter_replicated_at_leaf_buffers) { in set_parameter_replicated_at_leaf_buffers() argument3375 parameter_replicated_at_leaf_buffers); in set_parameter_replicated_at_leaf_buffers()3379 HloInstruction::parameter_replicated_at_leaf_buffers() const { in parameter_replicated_at_leaf_buffers() function in xla::HloInstruction3381 ->parameter_replicated_at_leaf_buffers(); in parameter_replicated_at_leaf_buffers()