Home
last modified time | relevance | path

Searched refs:SetDeviceOrdinalAttributeForNode (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.h52 Status SetDeviceOrdinalAttributeForNode(Node* node, int device_ordinal);
Dside_effect_util.cc37 Status SetDeviceOrdinalAttributeForNode(Node* node, int device_ordinal) { in SetDeviceOrdinalAttributeForNode() function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc303 TF_RETURN_IF_ERROR(SetDeviceOrdinalAttributeForNode(node, device.id)); in SetNodeDeviceForTPUCommunication()