[xla:ffi] Add support for array attributes to custom call thunk
PiperOrigin-RevId: 618878136
Showing
- tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc 0 additions, 1 deletion...ir/tensorflow/translate/tf_mlir_translate_registration.cc
- tensorflow/compiler/mlir/tf2xla/api/v1/BUILD 12 additions, 0 deletionstensorflow/compiler/mlir/tf2xla/api/v1/BUILD
- tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc 15 additions, 6 deletionstensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
- tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h 6 additions, 3 deletionstensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
- tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc 70 additions, 0 deletions...low/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
- tensorflow/compiler/tf2xla/mlir_xla_op_kernel.cc 18 additions, 6 deletionstensorflow/compiler/tf2xla/mlir_xla_op_kernel.cc
- third_party/xla/xla/ffi/BUILD 1 addition, 0 deletionsthird_party/xla/xla/ffi/BUILD
- third_party/xla/xla/ffi/api/api.h 4 additions, 1 deletionthird_party/xla/xla/ffi/api/api.h
- third_party/xla/xla/ffi/ffi_test.cc 12 additions, 2 deletionsthird_party/xla/xla/ffi/ffi_test.cc
- third_party/xla/xla/service/gpu/BUILD 1 addition, 0 deletionsthird_party/xla/xla/service/gpu/BUILD
- third_party/xla/xla/service/gpu/custom_call_test.cc 36 additions, 0 deletionsthird_party/xla/xla/service/gpu/custom_call_test.cc
- third_party/xla/xla/service/gpu/runtime/BUILD 3 additions, 0 deletionsthird_party/xla/xla/service/gpu/runtime/BUILD
- third_party/xla/xla/service/gpu/runtime/custom_call_thunk.cc 20 additions, 0 deletionsthird_party/xla/xla/service/gpu/runtime/custom_call_thunk.cc
Please register or sign in to comment