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 | 585 Status GetStepMarkerLocation(const Node& replicate_node, in GetStepMarkerLocation()
2157 bool UseSpmdForXlaPartitioning(const Node* replicate_node) { in UseSpmdForXlaPartitioning()
2189 const Node* replicate_node, FunctionLibraryRuntime* flr, in AssignArgsAndRetvalsToCores()
2516 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes, in BuildDynamicShapeNodes()
2646 const Node* replicate_node, const NameAttrList& function, in BuildCompileNode()
3214 const Node& replicate_node, const bool enable_xla_param_broadcast, in CreateOrGetPerHostVariableCopy()
3301 const Node& replicate_node, const std::vector<std::string>& arg_names, in BuildExecuteNodes()
4406 const Node& replicate_node, in ParseHostComputeCores()
4425 const DeviceSet& device_set, const Node& replicate_node, int* num_replicas, in GetDeviceTopology()
4480 int num_replicas, const Node& replicate_node, FunctionLibraryRuntime* flr, in GetIOTypes()
[all …]
|