- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 198 for _xla (0.04 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/mangling_util.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/utils.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/translate/hlo_to_mhlo/attribute_importer.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/numeric_types.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.h" #include "tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/xla.pb.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/framework/types.pb.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/stablehlo/BUILD
load("@local_xla//xla/tsl:tsl.default.bzl", "tsl_pybind_extension") load("//tensorflow:pytype.default.bzl", "pytype_strict_library") load("//tensorflow:strict.default.bzl", "py_strict_test") package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], default_visibility = [ ":friends", "//tensorflow/tools/pip_package:__subpackages__", ], licenses = ["notice"], ) package_group(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 18:15:11 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
llvm::cl::desc("Choose target opset."), llvm::cl::values( clEnumValN(OpSet::TF, "TF", "Uses TF ops that mimic quantization behavior"), clEnumValN(OpSet::XLA, "XLA", "Uses TF XLA ops"), clEnumValN(OpSet::UNIFORM_QUANTIZED, "UNIFORM_QUANTIZED", "Uses TF Uniform Quantized ops"))}; Option<QuantMethod> quantization_method_{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
limitations under the License. ==============================================================================*/ // This transformation pass converts stateful and stateless partitioned calls // with _xla_compile_device_type attribute to XLA launch ops. #include <stack> #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// RUN: tf-opt %s -split-input-file -tf-xla-broadcast | FileCheck %s
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
} def XlaBroadcastPass : Pass<"tf-xla-broadcast", "mlir::func::FuncOp"> { let summary = "Moves a broadcast from host into XLA, encoded as XlaAllReduce"; let description = [{ This pass moves brodcasts from host TF ops into XLA. This enables use of the inter-device network, which is faster than the inter-host network. Broadcasts in XLA are encoded as XlaAllReduce. An all_reduce with all 0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/BUILD
"@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:Pass", "@llvm-project//mlir:ReconcileUnrealizedCasts", "@llvm-project//mlir:Transforms", "@local_xla//xla/mlir_hlo:mhlo_passes", ], ) cc_library( name = "pre_calibration", srcs = ["pre_calibration.cc"], hdrs = ["pre_calibration.h"], compatible_with = get_compatible_with_portable(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 17K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.h
#include "mlir/IR/BuiltinTypeInterfaces.h" // from @llvm-project #include "mlir/Transforms/DialectConversion.h" // from @llvm-project #include "tensorflow/compiler/mlir/lite/transforms/passes.h" // IWYU pragma: keep #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" // IWYU pragma: keep namespace mlir { namespace odml { // Given a Composite op that wraps a core.aten.avg_pool2d, returns the padding
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.h
#include "mlir/IR/BuiltinAttributes.h" // from @llvm-project #include "mlir/IR/Types.h" // from @llvm-project #include "mlir/IR/Value.h" // from @llvm-project #include "tensorflow/compiler/mlir/lite/schema/schema_generated.h" #include "xla/statusor.h" #include "tensorflow/core/framework/types.pb.h" namespace mlir { class Builder; } // namespace mlir namespace tflite { // Convert the MLIR type to the corresponding TFLite tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 2.1K bytes - Viewed (0)