HomeSort by relevance Sort by last modified time
    Searched refs:has_sharding (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_tfgraph_builder.cc 105 instruction->has_sharding() &&
218 if (instruction->has_sharding() &&
batchnorm_expander.cc 218 if (use_fusion_ && !batch_norm->has_sharding()) {
285 if (batch_norm->has_sharding()) {
387 if (batch_norm->has_sharding()) {
514 if (use_fusion_ && !batch_norm->has_sharding()) {
579 if (batch_norm->has_sharding()) {
user_computation_test.cc 268 EXPECT_TRUE(broadcast->has_sharding());
271 EXPECT_TRUE(reshape->has_sharding());
hlo_instruction.h 937 bool has_sharding() const { return sharding_ != nullptr; } function in class:xla::HloInstruction
    [all...]
hlo_instruction.cc 689 if (operand->has_sharding()) {
714 if (operand->has_sharding()) {
721 if (operand->has_sharding()) {
    [all...]
hlo_graph_dumper.cc 850 if (!instr->has_sharding()) {
    [all...]
service.cc     [all...]
user_computation.cc     [all...]

Completed in 149 milliseconds