Searched refs:GetIndexAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 69 Status GetIndexAttr(const Node& n, int num_args, int* index) { in GetIndexAttr() function
159 TF_RETURN_IF_ERROR(GetIndexAttr(*args[i], num_args, &index)); in RewriteSubgraph()
173 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 60 Status GetIndexAttr(const Node& n, int num_args, int* index) { in GetIndexAttr() function
86 TF_CHECK_OK(GetIndexAttr(n, arg_source_tensors.size(), &index)); in RewriteSubgraph()
102 TF_CHECK_OK(GetIndexAttr(n, arg_source_tensors.size(), &index)); in RewriteSubgraph()
218 TF_RETURN_IF_ERROR(GetIndexAttr(*args[i], num_args, &index)); in RewriteSubgraph()
236 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()
|