OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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 258 milliseconds