Searched defs:replicate_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 569 const Node& replicate_node, in FillPaddingMap()
585 Status GetStepMarkerLocation(const Node& replicate_node, in GetStepMarkerLocation()
1885 bool UseSpmdForXlaPartitioning(const Node* replicate_node) { in UseSpmdForXlaPartitioning()
1917 const Node* replicate_node, FunctionLibraryRuntime* flr, in AssignArgsAndRetvalsToCores()
2234 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes, in BuildDynamicShapeNodes()
2305 const Node* replicate_node, const NameAttrList& function, in BuildCompileNode()
2794 const Node& replicate_node, const bool enable_xla_param_broadcast, in CreateOrGetPerHostVariableCopy()
2888 const Node& replicate_node, const std::vector<std::string>& arg_names, in BuildExecuteNodes()
3899 const Node& replicate_node, in ParseHostComputeCores()
3918 const DeviceSet& device_set, const Node& replicate_node, int* num_replicas, in GetDeviceTopology()
[all …]
|