| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores_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()
|
| D | cudnn_conv_rewriter_test.cc | 129 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
165 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
195 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
225 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
254 op::CustomCall(kCudnnConvBackwardFilterCallTarget), 0)); in TEST_F()
308 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
355 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
389 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
443 op::CustomCall(kCudnnConvBackwardInputCallTarget), 0)); in TEST_F()
493 op::GetTupleElement(op::CustomCall(kCudnnConvForwardCallTarget), 0)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers_test.cc | 86 EXPECT_THAT(call.get(), op::CustomCall()); in TEST()
87 EXPECT_THAT(call.get(), op::CustomCall(c1.get(), c2.get())); in TEST()
88 EXPECT_THAT(call.get(), op::CustomCall("foo_target")); in TEST()
89 EXPECT_THAT(call.get(), op::CustomCall("foo_target", c1.get(), c2.get())); in TEST()
90 EXPECT_THAT(call.get(), op::CustomCall(::testing::StartsWith("foo"))); in TEST()
92 op::CustomCall(::testing::Not(::testing::StartsWith("bar")))); in TEST()
95 EXPECT_THAT(call.get(), ::testing::Not(op::CustomCall(c1.get()))); in TEST()
99 ::testing::Not(op::CustomCall(::testing::StartsWith("bar")))); in TEST()
101 EXPECT_THAT(Explain(call.get(), op::CustomCall("bar")), in TEST()
103 EXPECT_THAT(DescribeHloMatcher(op::CustomCall("foo_target")), in TEST()
|
| D | hlo_matchers.h | 327 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function
339 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall( in CustomCall() function
344 inline ::testing::Matcher<const ::xla::HloInstruction*> CustomCall() { in CustomCall() function
|
| D | layout_assignment_test.cc | 1159 ASSERT_THAT(root, GmockMatch(m::CustomCall(m::Parameter()))); in TEST_F()
1175 ASSERT_THAT(root, GmockMatch(m::CustomCall(m::Parameter()))); in TEST_F()
1206 GmockMatch(m::Copy(m::CustomCall(m::Copy(), m::Parameter())))); in TEST_F()
1232 GmockMatch(m::Copy(m::CustomCall()))); in TEST_F()
1266 GmockMatch(m::Copy(m::CustomCall(m::Tuple())))); in TEST_F()
|
| D | hlo_parser_test.cc | 1280 "CustomCall", in CreateTestCases()
1283 ENTRY CustomCall { in CreateTestCases()
1295 ENTRY CustomCall { in CreateTestCases()
2035 "Shape of computation CustomCall, f32[1], is not compatible "
|
| D | pattern_matcher.h | 2276 XLA_VARIADIC_OP_PATTERN(CustomCall);
|
| /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/python/ |
| D | local_computation_builder.h | 311 LocalOp CustomCall(const string& call_target_name,
|
| D | local_computation_builder.i | 400 %unignore xla::swig::ComputationBuilder::CustomCall;
|
| D | xla_client.py | 1418 def CustomCall(self, member in ComputationBuilder
1439 return self._client.CustomCall(call_target_name, operands,
|
| D | local_computation_builder.cc | 618 LocalOp ComputationBuilder::CustomCall( in CustomCall() function in xla::swig::ComputationBuilder
|
| D | xla_client_test.py | 319 c.CustomCall(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 435 XlaOp CustomCall(
824 friend XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name,
1431 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name,
|
| D | xla_builder.cc | 1493 XlaOp XlaBuilder::CustomCall( in CustomCall() function in xla::XlaBuilder
3081 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name, in CustomCall() function
3084 return builder->CustomCall(call_target_name, operands, shape, opaque, in CustomCall()
3092 return builder->CustomCall(call_target_name, operands, shape, opaque, in CustomCallWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 816 ## CustomCall section in Operation Semantics
819 [`XlaBuilder::CustomCall`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
823 <b> `CustomCall(target_name, args..., shape)` </b>
841 For example, if CustomCall is used as follows:
847 CustomCall("myfunc", {x, y}, f32[3x3])
875 > native XLA ops whenever possible; only use CustomCall as a last resort.
|