HomeSort by relevance Sort by last modified time
    Searched full:set_sharding (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
batchnorm_expander.cc 293 inst->set_sharding(operand_sharding);
295 inst->set_sharding(HloSharding::Replicate());
298 tuple->set_sharding(batch_norm->sharding());
390 inst->set_sharding(batch_norm->sharding());
392 inst->set_sharding(HloSharding::Replicate());
395 shifted_normalized->set_sharding(batch_norm->sharding());
587 inst->set_sharding(activation_sharding);
589 inst->set_sharding(HloSharding::Replicate());
592 tuple->set_sharding(batch_norm->sharding());
hlo_instruction.cc 690 broadcast->set_sharding(operand->sharding());
715 reshaped_operand->set_sharding(operand->sharding());
722 broadcast->set_sharding(operand->sharding());
    [all...]
hlo_instruction.h 931 void set_sharding(const HloSharding& sharding) { function in class:xla::HloInstruction
    [all...]
user_computation.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc     [all...]

Completed in 86 milliseconds