Searched refs:CustomCallWithLayout (Results 1 – 8 of 8) sorted by relevance
180 return CustomCallWithLayout( in BuildOpsSubmodule()201 return CustomCallWithLayout( in BuildOpsSubmodule()
678 CustomCallWithLayout(&builder, callback_name, custom_call_args, in EmitPythonCallbackFromDescriptor()
372 ops.CustomCallWithLayout(397 ops.CustomCallWithLayout(
217 xla::XlaOp out = xla::CustomCallWithLayout( in CompileToCustomCallCallingTfKernel()
182 def CustomCallWithLayout(
1275 friend XlaOp CustomCallWithLayout(2158 XlaOp CustomCallWithLayout(
4449 XlaOp CustomCallWithLayout( in CustomCallWithLayout() function
1136 value_map[result] = xla::CustomCallWithLayout( in ExportXlaOp()