Searched defs:params_info (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1694 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes, in GetArgAndRetvalShapes()
1912 int num_cores_per_replica, const ParameterInfo& params_info, in AssignArgsAndRetvalsToCores()
2235 const ParameterInfo& params_info, const std::vector<Node*>& variable_reads, in BuildDynamicShapeNodes()
2306 uint64 library_fingerprint, const ParameterInfo& params_info, in BuildCompileNode()
2792 const DistributedTPURewritePass::ParameterInfo& params_info, in CreateOrGetPerHostVariableCopy()
2887 const ParameterInfo& params_info, int num_tasks, int num_cores_per_replica, in BuildExecuteNodes()
3975 DataTypeVector* retval_types, ParameterInfo* params_info) { in GetIOTypes()
4247 ParameterInfo params_info; in RewriteTPUReplicateNode() local
|