| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions_test.cc | 50 EXPECT_THAT(root, op::CustomCall(kCudnnConvForwardCallTarget, in TEST_F()
73 EXPECT_THAT(root, op::CustomCall(kCudnnConvBackwardInputCallTarget, in TEST_F()
96 EXPECT_THAT(root, op::Tuple(op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
118 op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
140 op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
162 op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
184 EXPECT_THAT(root, op::CustomCall(kCudnnConvForwardCallTarget, in TEST_F()
209 EXPECT_THAT(root, op::CustomCall(kCudnnConvForwardCallTarget, in TEST_F()
233 EXPECT_THAT(root, op::Tuple(op::Slice(op::GetTupleElement(op::CustomCall( in TEST_F()
|
| D | custom_call_test.cc | 41 CustomCall(&b, "Callback_IsInvoked", /*operands=*/{}, in TEST_F()
51 CustomCall(&b, "UnknownTarget", /*operands=*/{}, in TEST_F()
68 CustomCall(&b, "Callback_Memcpy", in TEST_F()
85 CustomCall(&b, "Callback_Opaque", /*operands=*/{}, in TEST_F()
118 CustomCall(&b, "Callback_SubBuffers", /*operands=*/ in TEST_F()
175 auto cc = CustomCall(&b, "Callback_TupleSelect", in TEST_F()
|
| D | gpu_conv_rewriter_test.cc | 129 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
183 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
245 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
275 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
305 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
334 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
388 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
435 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
469 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
523 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers_test.cc | 99 EXPECT_THAT(call.get(), op::CustomCall()); in TEST_F()
100 EXPECT_THAT(call.get(), op::CustomCall(c1.get(), c2.get())); in TEST_F()
101 EXPECT_THAT(call.get(), op::CustomCall("foo_target")); in TEST_F()
102 EXPECT_THAT(call.get(), op::CustomCall("foo_target", c1.get(), c2.get())); in TEST_F()
103 EXPECT_THAT(call.get(), op::CustomCall(::testing::StartsWith("foo"))); in TEST_F()
105 op::CustomCall(::testing::Not(::testing::StartsWith("bar")))); in TEST_F()
108 EXPECT_THAT(call.get(), ::testing::Not(op::CustomCall(c1.get()))); in TEST_F()
112 ::testing::Not(op::CustomCall(::testing::StartsWith("bar")))); in TEST_F()
114 EXPECT_THAT(Explain(call.get(), op::CustomCall("bar")), in TEST_F()
118 EXPECT_THAT(DescribeHloMatcher(op::CustomCall("foo_target")), in TEST_F()
|
| D | hlo_matchers.h | 355 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function
367 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function
372 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall() { in CustomCall() function
|
| D | layout_assignment_test.cc | 1153 ASSERT_THAT(root, GmockMatch(m::CustomCall(m::Parameter()))); in TEST_F()
1169 ASSERT_THAT(root, GmockMatch(m::CustomCall(m::Parameter()))); in TEST_F()
1200 GmockMatch(m::Copy(m::CustomCall(m::Copy(), m::Parameter())))); in TEST_F()
1226 GmockMatch(m::Copy(m::CustomCall()))); in TEST_F()
1260 GmockMatch(m::Copy(m::CustomCall(m::Tuple())))); in TEST_F()
|
| D | hlo_parser_test.cc | 396 ENTRY %CustomCall () -> f32[1,2,3] {
1351 "CustomCall",
1354 ENTRY CustomCall {
|
| D | pattern_matcher.h | 2289 XLA_VARIADIC_OP_PATTERN(CustomCall);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sharding_op.cc | 37 0, xla::CustomCall(ctx->builder(), /*call_target_name=*/"Sharding", in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_aot_test_helper.cc | 56 auto sum = CustomCall(&builder, "SumStructElements", {opaque_param}, r0f32); in main()
|
| D | custom_call_test.cc | 216 CustomCall(&builder, "$illegal", /*operands=*/{}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | custom_call.md | 37 xla::CustomCall(&b, "do_custom_call", /*operands=*/{param0, param1},
111 xla::CustomCall(&b, "do_custom_call", /*operands=*/{param0, param1},
142 xla::CustomCall(&b, "do_custom_call", /*operands=*/{p0}, out_shape);
|
| D | operation_semantics.md | 852 ## CustomCall section in Operation Semantics
855 [`XlaBuilder::CustomCall`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
859 <b> `CustomCall(target_name, args..., shape)` </b>
877 For example, if CustomCall is used as follows:
883 CustomCall("myfunc", {x, y}, f32[3x3])
911 > native XLA ops whenever possible; only use CustomCall as a last resort.
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1195 def CustomCall(self, member in ComputationBuilder
1216 return ops.CustomCall(self._builder, call_target_name,
|
| D | xla_client_test.py | 342 c.CustomCall(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 481 XlaOp CustomCall(
863 friend XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name,
1519 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name,
|
| D | xla_builder.cc | 1549 XlaOp XlaBuilder::CustomCall( in CustomCall() function in xla::XlaBuilder
3146 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name, in CustomCall() function
3149 return builder->CustomCall(call_target_name, operands, shape, opaque, in CustomCall()
3157 return builder->CustomCall(call_target_name, operands, shape, opaque, in CustomCallWithLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 553 value_map[op] = xla::CustomCall( in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops_base.td | 888 string summary = "CustomCall operator";
|