Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening_test.cc265 EXPECT_THAT(tuple, op::Tuple(op::CustomCall(), op::AfterAll())); in TEST_F()
287 EXPECT_THAT(tuple, op::Tuple(op::CustomCall(), op::AfterAll())); in TEST_F()
310 EXPECT_THAT(custom_call, op::CustomCall(op::Parameter(0), op::AfterAll())); in TEST_F()
521 op::Tuple(op::CustomCall(), op::AfterAll())); in TEST_F()
552 op::Tuple(op::CustomCall(), op::AfterAll())); in TEST_F()
581 op::CustomCall(op::Constant(), op::AfterAll())); in TEST_F()
613 op::CustomCall(op::Constant(), op::AfterAll())); in TEST_F()
/external/javaparser/javaparser-symbol-solver-testing/src/test/java/com/github/javaparser/symbolsolver/resolution/
DAnonymousClassesResolutionTest.java12 import org.junit.jupiter.api.AfterAll;
39 @AfterAll
DAnnotationsResolutionTest.java20 import org.junit.jupiter.api.AfterAll;
46 @AfterAll
/external/apache-commons-io/src/test/java/org/apache/commons/io/output/
DQueueOutputStreamTest.java36 import org.junit.jupiter.api.AfterAll;
48 @AfterAll
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dlocation_to_op_metadata.mlir17 %0 = "mhlo.after_all"(%arg0) : (!mhlo.token) -> !mhlo.token loc("AfterAll")
22 // CHECK-SAME: metadata={op_name="AfterAll"}
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_tfrt_cpu_device_buffer.cc33 tfrt::AsyncValueRef<CpuEvent> AfterAll( in AfterAll() function
79 AfterAll(definition_events), in TrackedTfrtCpuDeviceBuffer()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc157 xla::XlaOp token = xla::AfterAll(b, input_tokens); in Compile()
185 : xla::AfterAll(b, send_to_host_tokens); in Compile()
253 : xla::AfterAll(b, recv_from_host_tokens); in Compile()
438 xla::XlaOp token = xla::AfterAll(b, input_tokens); in Compile()
496 xla::XlaOp token = xla::AfterAll(b, input_tokens); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_simplifier_test.cc514 op::Tuple(op::AfterAll( in TEST_F()
515 op::GetTupleElement(op::Tuple(op::AfterAll()), 0), in TEST_F()
516 op::GetTupleElement(op::Tuple(op::AfterAll()), 0)))); in TEST_F()
Dhlo_matchers.h239 HLO_MATCHER(AfterAll);
Dhlo_computation_test.cc376 EXPECT_THAT(copy, GmockMatch(m::AfterAll())); in TEST_F()
/external/javaparser/javaparser-core-serialization/src/test/java/com/github/javaparser/serialization/
DJavaParserJsonDeserializerTest.java35 import org.junit.jupiter.api.AfterAll;
209 @AfterAll
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc294 xla::XlaOp token_input = xla::AfterAll(b, token_inputs); in CompileFunctionalNode()
Dxla_compiler.cc1407 token_output.reset(new xla::XlaOp(xla::AfterAll(&builder, token_inputs))); in CompileGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcase_op.cc288 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dif_op.cc308 inputs[i] = xla::AfterAll(b, token_inputs); in Compile()
Dwhile_op.cc510 inputs[i] = xla::AfterAll(builder, token_inputs); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc372 op::AfterAll(), op::AfterAll())))); in TEST_F()
402 op::Convert(op::PartitionId()), op::AfterAll(), in TEST_F()
403 op::AfterAll())))); in TEST_F()
437 op::Convert(op::PartitionId()), op::AfterAll(), in TEST_F()
438 op::AfterAll())))); in TEST_F()
713 op::Tuple(op::Parameter(0), op::AfterAll()), in TEST_F()
714 op::Tuple(op::Parameter(0), op::AfterAll())))); in TEST_F()
723 EXPECT_THAT(root_b1, AllOf(op::Shape("token[]"), op::AfterAll())); in TEST_F()
740 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
760 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
Dops.pyi63 def AfterAll(builder: XlaBuilder, tokens: Sequence[XlaOp]) -> XlaOp: ...
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc72 ops.def("AfterAll", &AfterAll, py::arg("builder"), py::arg("tokens")); in BuildOpsSubmodule()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td113 This is exported to HLO as an AfterAll operation with no operands to
862 the operation is the data operand. When used with AfterAll this operation
883 let summary = "AfterAll operator";
886 AfterAll takes a variadic number of tokens and produces a single token.
888 operations to enforce ordering. AfterAll can be used as a join of tokens
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.td115 This is exported to HLO as an AfterAll operation with no operands to
828 let summary = "AfterAll operator";
831 AfterAll takes a variadic number of tokens and produces a single token.
833 operations to enforce ordering. AfterAll can be used as a join of tokens
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h910 XlaOp AfterAll(absl::Span<const XlaOp> tokens);
1509 friend XlaOp AfterAll(XlaBuilder* builder, absl::Span<const XlaOp> tokens);
2660 XlaOp AfterAll(XlaBuilder* builder, absl::Span<const XlaOp> tokens);
Dxla_builder_test.cc1233 AfterAll(&b, {CreateToken(&b), ConstantR0<float>(&b, 1.0)}); in TEST_F()
Dxla_builder.cc1945 XlaOp XlaBuilder::AfterAll(absl::Span<const XlaOp> tokens) { in AfterAll() function in xla::XlaBuilder
4969 XlaOp AfterAll(XlaBuilder* builder, absl::Span<const XlaOp> tokens) { in AfterAll() function
4970 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>

12