Searched refs:CustomCallWithLayout (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops_cpu.cc | 121 output = xla::CustomCallWithLayout(&b, "argmax_float_1d_xla_impl", args, in Compile()
124 output = xla::CustomCallWithLayout(&b, "argmax_float_2d_xla_impl", args, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 827 friend XlaOp CustomCallWithLayout(
1440 XlaOp CustomCallWithLayout(XlaBuilder* builder, const string& call_target_name,
|
| D | xla_builder.cc | 3088 XlaOp CustomCallWithLayout(XlaBuilder* builder, const string& call_target_name, in CustomCallWithLayout() function
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 628 return xla::CustomCallWithLayout(&builder_, call_target_name, xla_ops, in CustomCall()
|