Searched defs:replicate_node (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 588 Status GetStepMarkerLocation(const Node& replicate_node, in GetStepMarkerLocation()
2098 bool UseSpmdForXlaPartitioning(const Node* replicate_node) { in UseSpmdForXlaPartitioning()
2130 const Node* replicate_node, FunctionLibraryRuntime* flr, in AssignArgsAndRetvalsToCores()
2464 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes, in BuildDynamicShapeNodes()
2593 const Node* replicate_node, const NameAttrList& function, in BuildCompileNode()
3156 const Node& replicate_node, const bool enable_xla_param_broadcast, in CreateOrGetPerHostVariableCopy()
3241 const Node& replicate_node, const std::vector<std::string>& arg_names, in BuildExecuteNodes()
4353 const Node& replicate_node, in ParseHostComputeCores()
4372 const DeviceSet& device_set, const Node& replicate_node, int* num_replicas, in GetDeviceTopology()
4427 int num_replicas, const Node& replicate_node, FunctionLibraryRuntime* flr, in GetIOTypes()
[all …]
|