Home
last modified time | relevance | path

Searched refs:SetIsDistributedCommunication (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dcollective_ops.cc36 .SetIsDistributedCommunication()
50 .SetIsDistributedCommunication()
95 .SetIsDistributedCommunication()
108 .SetIsDistributedCommunication()
120 .SetIsDistributedCommunication()
142 .SetIsDistributedCommunication()
157 .SetIsDistributedCommunication()
181 .SetIsDistributedCommunication()
195 .SetIsDistributedCommunication()
213 .SetIsDistributedCommunication()
[all …]
H A Dsendrecv_ops.cc65 .SetIsDistributedCommunication()
90 .SetIsDistributedCommunication()
129 .SetIsDistributedCommunication()
H A Dbatch_ops.cc60 .SetIsDistributedCommunication();
93 .SetIsDistributedCommunication();
H A Dfunctional_ops.cc295 .SetIsDistributedCommunication()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ops/
H A Dtpu_execute_op.cc29 .SetIsDistributedCommunication()
50 .SetIsDistributedCommunication()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop.h278 OpDefBuilderWrapper& SetIsDistributedCommunication() { in SetIsDistributedCommunication() function
279 builder_.SetIsDistributedCommunication(); in SetIsDistributedCommunication()
H A Dop_def_builder.h202 OpDefBuilder& SetIsDistributedCommunication();
H A Dop_def_builder.cc618 OpDefBuilder& OpDefBuilder::SetIsDistributedCommunication() { in SetIsDistributedCommunication() function in tensorflow::OpDefBuilder
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/preemption/
H A Dcheck_preemption_op.cc24 .SetIsDistributedCommunication()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_test.cc969 .SetIsDistributedCommunication()