Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc287 std::vector<xla::XlaOp> token_inputs; in CompileFunctionalNode() local
291 token_inputs.push_back(token_or.ConsumeValueOrDie()); in CompileFunctionalNode()
293 xla::XlaOp token_input = xla::AfterAll(b, token_inputs); in CompileFunctionalNode()
Dxla_compiler.cc1368 std::vector<xla::XlaOp> token_inputs; in CompileGraph() local
1372 token_inputs.push_back(token_or.ValueOrDie()); in CompileGraph()
1374 token_output.reset(new xla::XlaOp(xla::AfterAll(&builder, token_inputs))); in CompileGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcase_op.cc290 std::vector<xla::XlaOp> token_inputs; in Compile() local
294 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
296 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dif_op.cc278 std::vector<xla::XlaOp> token_inputs; in Compile() local
282 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
284 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dwhile_op.cc491 std::vector<xla::XlaOp> token_inputs; in Compile() local
495 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
497 inputs[i] = xla::AfterAll(builder, token_inputs); in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc571 std::vector<string> token_inputs; in TEST_F() local
573 GetNodeAttr(if_node->def(), "_xla_token_input_nodes", &token_inputs)); in TEST_F()
574 EXPECT_THAT(token_inputs, ::testing::ElementsAre("send_oc_if_pred_if")); in TEST_F()