Home
last modified time | relevance | path

Searched refs:replicated_at_leaf_buffers (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h444 const std::vector<bool>& replicated_at_leaf_buffers);
1054 const std::vector<bool>& replicated_at_leaf_buffers);
1560 const std::vector<bool>& replicated_at_leaf_buffers);
Dxla_builder.cc846 const std::vector<bool>& replicated_at_leaf_buffers) { in Parameter() argument
856 if (!replicated_at_leaf_buffers.empty()) { in Parameter()
858 for (bool replicated : replicated_at_leaf_buffers) { in Parameter()
4087 const std::vector<bool>& replicated_at_leaf_buffers) { in Parameter() argument
4089 replicated_at_leaf_buffers); in Parameter()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2621 parameter_replication->replicated_at_leaf_buffers(); in ParseInstructionRhs()
5132 parameter_replication.replicated_at_leaf_buffers().begin(), in ParseParameterReplicationOnly()
5133 parameter_replication.replicated_at_leaf_buffers().end()); in ParseParameterReplicationOnly()
Dhlo_instruction.cc382 if (!proto.parameter_replication().replicated_at_leaf_buffers().empty()) { in CreateFromProto()
384 proto.parameter_replication().replicated_at_leaf_buffers()); in CreateFromProto()
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto712 repeated bool replicated_at_leaf_buffers = 1; field