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 | 1907 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes, in GetArgAndRetvalShapes()
2125 int num_cores_per_replica, const ParameterInfo& params_info, in AssignArgsAndRetvalsToCores()
2465 const ParameterInfo& params_info, const std::vector<Node*>& variable_reads, in BuildDynamicShapeNodes()
2563 const DistributedTPURewritePass::ParameterInfo& params_info, in XlaBroadcastKindSupported()
2577 const DistributedTPURewritePass::ParameterInfo& params_info, int param_num, in EnableXlaParamBroadcast()
2594 uint64 library_fingerprint, const ParameterInfo& params_info, in BuildCompileNode()
3154 const DistributedTPURewritePass::ParameterInfo& params_info, in CreateOrGetPerHostVariableCopy()
3240 const ParameterInfo& params_info, int num_tasks, int num_cores_per_replica, in BuildExecuteNodes()
4430 DataTypeVector* retval_types, ParameterInfo* params_info) { in GetIOTypes()
4749 ParameterInfo params_info; in RewriteTPUReplicateNode() local
|