Home
last modified time | relevance | path

Searched refs:AfterAll (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc282 xla::XlaOp token_input = xla::AfterAll(b, token_inputs); in CompileFunctionalNode()
Dxla_compiler.cc1137 token_output.reset(new xla::XlaOp(xla::AfterAll(&builder, token_inputs))); in CompileGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc212 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dcase_op.cc224 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dwhile_op.cc477 inputs[i] = xla::AfterAll(builder, token_inputs); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.h206 HLO_MATCHER(AfterAll);
Dhlo_computation_test.cc395 EXPECT_THAT(copy, GmockMatch(m::AfterAll())); in TEST_F()
Dpattern_matcher.h2274 XLA_VARIADIC_OP_PATTERN(AfterAll);
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h563 XlaOp AfterAll(absl::Span<const XlaOp> tokens);
1004 friend XlaOp AfterAll(XlaBuilder* builder, absl::Span<const XlaOp> tokens);
1875 XlaOp AfterAll(XlaBuilder* builder, absl::Span<const XlaOp> tokens);
Dxla_builder_test.cc948 AfterAll(&b, {CreateToken(&b), ConstantR0<float>(&b, 1.0)}); in TEST_F()
Dxla_builder.cc1473 XlaOp XlaBuilder::AfterAll(absl::Span<const XlaOp> tokens) { in AfterAll() function in xla::XlaBuilder
3482 XlaOp AfterAll(XlaBuilder* builder, absl::Span<const XlaOp> tokens) { in AfterAll() function
3483 return builder->AfterAll(tokens); in AfterAll()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md16 ## AfterAll section in Operation Semantics
19 [`XlaBuilder::AfterAll`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builder…
21 AfterAll takes a variadic number of tokens and produces a single token. Tokens
23 enforce ordering. `AfterAll` can be used as a join of tokens for ordering a
26 <b> `AfterAll(operands)` </b>