Searched refs:token_input_nodes_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 119 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in HostComputeOp()
120 OP_REQUIRES(ctx, !token_input_nodes_.empty(), in HostComputeOp()
140 for (auto& token_input_node : token_input_nodes_) { in Compile()
384 std::vector<string> token_input_nodes_; member in tensorflow::__anon4d7735cb0111::HostComputeOp
395 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in SendToHostOp()
396 OP_REQUIRES(ctx, !token_input_nodes_.empty(), in SendToHostOp()
411 for (auto& token_input_node : token_input_nodes_) { in Compile()
438 std::vector<string> token_input_nodes_; member in tensorflow::__anon4d7735cb0111::SendToHostOp
450 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in RecvFromHostOp()
451 OP_REQUIRES(ctx, !token_input_nodes_.empty(), in RecvFromHostOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.h | 60 std::vector<string> token_input_nodes_; variable
|
| D | if_op.h | 56 std::vector<string> token_input_nodes_; variable
|
| D | case_op.h | 66 std::vector<string> token_input_nodes_; variable
|
| D | if_op.cc | 39 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaIfOp()
42 has_token_input_output_ = !token_input_nodes_.empty(); in XlaIfOp()
279 for (const string& node_name : token_input_nodes_) { in Compile()
|
| D | case_op.cc | 34 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaCaseOp()
37 has_token_input_output_ = !token_input_nodes_.empty(); in XlaCaseOp()
291 for (const string& node_name : token_input_nodes_) { in Compile()
|
| D | while_op.cc | 278 if (!ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_).ok()) { in XlaWhileOp()
281 has_token_input_output_ = !token_input_nodes_.empty(); in XlaWhileOp()
492 for (const string& node_name : token_input_nodes_) { in Compile()
|