Home
last modified time | relevance | path

Searched refs:token_or (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc141 auto token_or = compiler->GetNodeToken(token_input_node); in Compile() local
142 OP_REQUIRES_OK(ctx, token_or.status()); in Compile()
143 input_tokens.push_back(token_or.ValueOrDie()); in Compile()
412 auto token_or = compiler->GetNodeToken(token_input_node); in Compile() local
413 OP_REQUIRES_OK(ctx, token_or.status()); in Compile()
414 input_tokens.push_back(token_or.ValueOrDie()); in Compile()
464 auto token_or = compiler->GetNodeToken(token_input_node); in Compile() local
465 OP_REQUIRES_OK(ctx, token_or.status()); in Compile()
466 input_tokens.push_back(token_or.ValueOrDie()); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc289 auto token_or = compiler->GetNodeToken(node_name); in CompileFunctionalNode() local
290 TF_RETURN_IF_ERROR(token_or.status()); in CompileFunctionalNode()
291 token_inputs.push_back(token_or.ConsumeValueOrDie()); in CompileFunctionalNode()
Dxla_compiler.cc1370 auto token_or = GetNodeToken(node_name); in CompileGraph() local
1371 TF_RETURN_IF_ERROR(token_or.status()); in CompileGraph()
1372 token_inputs.push_back(token_or.ValueOrDie()); in CompileGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcase_op.cc292 auto token_or = compiler->GetNodeToken(node_name); in Compile() local
293 OP_REQUIRES_OK(ctx, token_or.status()); in Compile()
294 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
Dif_op.cc280 auto token_or = compiler->GetNodeToken(node_name); in Compile() local
281 OP_REQUIRES_OK(ctx, token_or.status()); in Compile()
282 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
Dwhile_op.cc493 auto token_or = compiler->GetNodeToken(node_name); in Compile() local
494 OP_REQUIRES_OK(ctx, token_or.status()); in Compile()
495 token_inputs.push_back(token_or.ValueOrDie()); in Compile()