- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for STABLEHLO_CUSTOM_CALL (0.46 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/stablehlo_custom_call.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/compiler/mlir/tensorflow/utils/stablehlo_custom_call.h" #include "mlir/IR/Attributes.h" // from @llvm-project #include "mlir/IR/BuiltinAttributes.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/utils/stablehlo_custom_call.h" #include "tensorflow/compiler/mlir/tensorflow/utils/visitor.h" #include "tensorflow/compiler/mlir/tensorflow/utils/xla_call_module_attrs.h" namespace mlir { namespace TF { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/utils/stablehlo_custom_call.h" #include "tensorflow/compiler/mlir/tensorflow/utils/xla_call_module_attrs.h" #include "tensorflow/compiler/tf2xla/kernels/xla_call_module_loader.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" // IWYU pragma: keep
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/BUILD
hdrs = ["utils/xla_call_module_attrs.h"], deps = ["@llvm-project//llvm:Support"], ) cc_library( name = "stablehlo_custom_call_utils", srcs = ["utils/stablehlo_custom_call.cc"], hdrs = ["utils/stablehlo_custom_call.h"], deps = [ ":xla_call_module_attrs", "@llvm-project//llvm:Support", "@llvm-project//mlir:IR", "@llvm-project//mlir:Support",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 47.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
return; } if (const auto* op = op_union.AsStablehloCustomCallOptions()) { // hard coding api version for now, we should rework this by updating the // STABLEHLO_CUSTOM_CALL definition attributes.emplace_back(builder.getNamedAttr( "api_version", mlir::vhlo::CustomCallApiVersionV1Attr::get( builder.getContext(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema.fbs
STABLEHLO_BROADCAST_IN_DIM = 170, // WARNING: No runtime support STABLEHLO_CONVOLUTION = 171, // WARNING: No runtime support STABLEHLO_SLICE = 172, // WARNING: No runtime support STABLEHLO_CUSTOM_CALL = 173, // WARNING: No runtime support STABLEHLO_REDUCE = 174, // WARNING: No runtime support STABLEHLO_ABS = 175, // WARNING: No runtime support STABLEHLO_AND = 176, // WARNING: No runtime support
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
"STABLEHLO_MAXIMUM", "STABLEHLO_RESHAPE", "STABLEHLO_CLAMP", "STABLEHLO_CONCATENATE", "STABLEHLO_BROADCAST_IN_DIM", "STABLEHLO_CONVOLUTION", "STABLEHLO_SLICE", "STABLEHLO_CUSTOM_CALL", "STABLEHLO_REDUCE", "STABLEHLO_ABS", "STABLEHLO_AND", "STABLEHLO_COSINE", "STABLEHLO_EXPONENTIAL", "STABLEHLO_FLOOR", "STABLEHLO_LOG", "STABLEHLO_MINIMUM",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)