Home
last modified time | relevance | path

Searched refs:CustomCallWithLayout (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dops.cc180 return CustomCallWithLayout( in BuildOpsSubmodule()
201 return CustomCallWithLayout( in BuildOpsSubmodule()
H A Dpy_client.cc678 CustomCallWithLayout(&builder, callback_name, custom_call_args, in EmitPythonCallbackFromDescriptor()
H A Dxla_client_test.py372 ops.CustomCallWithLayout(
397 ops.CustomCallWithLayout(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc217 xla::XlaOp out = xla::CustomCallWithLayout( in CompileToCustomCallCallingTfKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A Dops.pyi182 def CustomCallWithLayout(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h1275 friend XlaOp CustomCallWithLayout(
2158 XlaOp CustomCallWithLayout(
H A Dxla_builder.cc4449 XlaOp CustomCallWithLayout( in CustomCallWithLayout() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc1136 value_map[result] = xla::CustomCallWithLayout( in ExportXlaOp()