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.cc276 std::vector<xla::XlaOp> token_inputs; in CompileFunctionalNode() local
280 token_inputs.push_back(token_or.ConsumeValueOrDie()); in CompileFunctionalNode()
282 xla::XlaOp token_input = xla::AfterAll(b, token_inputs); in CompileFunctionalNode()
Dxla_compiler.cc1131 std::vector<xla::XlaOp> token_inputs; in CompileGraph() local
1135 token_inputs.push_back(token_or.ValueOrDie()); in CompileGraph()
1137 token_output.reset(new xla::XlaOp(xla::AfterAll(&builder, token_inputs))); in CompileGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc206 std::vector<xla::XlaOp> token_inputs; in Compile() local
210 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
212 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dcase_op.cc218 std::vector<xla::XlaOp> token_inputs; in Compile() local
222 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
224 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dwhile_op.cc471 std::vector<xla::XlaOp> token_inputs; in Compile() local
475 token_inputs.push_back(token_or.ValueOrDie()); in Compile()
477 inputs[i] = xla::AfterAll(builder, token_inputs); in Compile()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc685 std::vector<string> token_inputs; in TEST_F() local
687 GetNodeAttr(if_node->def(), "_xla_token_input_nodes", &token_inputs)); in TEST_F()
688 EXPECT_THAT(token_inputs, ::testing::ElementsAre("send_oc_if_pred_if")); in TEST_F()