- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 99 for _xla (0.04 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
return "quant-replace-cast-hacks-with-tf-xla-ops"; } StringRef getDescription() const final { // This is a brief description of the pass. return "Replace mixed-type Conv and Matmul cast hacks with TF XLA ops."; } void runOnOperation() override; }; // Generates params for the XLA Convolution op. void PrepareXlaConvParams(OpBuilder &builder, Location loc, ArrayAttr strides,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// Checks if a device string is a TPU replicated core device. bool IsTPUReplicatedCore(llvm::StringRef device); // Checks if `type` is allowed for XLA. String and resources are not XLA types. // There are other TF types that are not XLA types which will be removed by // successive passes in TF/XLA bridge phase 2. bool TypeValidForXLA(const mlir::Type& type); // Returns the map from core to the host that is associated with the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
} def XlaInlineDeviceOpsPass : Pass<"tf-xla-inline-device-ops", "ModuleOp"> { let summary = "Inline all Cluster op based in the parent region"; let constructor = "TFDevice::CreateXlaInlineDeviceOpsPass()"; let dependentDialects = ["tf_device::TensorFlowDeviceDialect"]; } def XlaRewritePass : Pass<"tf-xla-rewrite", "mlir::ModuleOp"> { let summary = "Rewrites partition calls into Xla launch ops to make the attached function run on XLA.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
#include "tensorflow/compiler/mlir/tf2xla/transforms/passes.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/mlir_hlo/mhlo/transforms/rewriters.h" #include "xla/mlir_hlo/mhlo/utils/type_conversion.h" #include "tensorflow/core/lib/monitoring/counter.h" namespace mlir { namespace mhlo { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h" #include "tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h" #include "xla/client/sharding_builder.h" #include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" namespace tensorflow { namespace tf2xla { namespace internal { namespace {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_types.h" #include "tensorflow/compiler/mlir/tensorflow/utils/device_util.h" #include "tensorflow/compiler/mlir/utils/string_container_utils.h" #include "xla/array4d.h" #include "xla/service/computation_placer.h" #include "xla/xla_data.pb.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/protobuf/tpu/topology.pb.h" #include "tensorflow/core/util/device_name_utils.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.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"))}; // Initialize for tests. void initializeForTest() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tf2xla/transforms/utils.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "xla/mlir_hlo/utils/convert_op_folder.h" #include "xla/mlir_hlo/utils/hlo_utils.h" #include "xla/xla_data.pb.h" namespace mlir { namespace mhlo { namespace { constexpr absl::string_view kGroupSizeAttrName =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/aot/tfcompile.bzl
"@local_xla//xla/service:hlo_profile_printer_data_cc", ] or []) + (include_standard_runtime_deps and [ # TODO(cwhipkey): only depend on kernel code that the model actually # needed. "@local_xla//xla/service/cpu:runtime_conv2d", "@local_xla//xla/service/cpu:runtime_custom_call_status", "@local_xla//xla/service/cpu:runtime_key_value_sort",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 19:18:08 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_passes.cc
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/passes.h" #include "tensorflow/compiler/mlir/quantization/tensorflow/quantization_options.pb.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" #include "xla/mlir_hlo/mhlo/transforms/passes.h" namespace tensorflow { namespace quantization { namespace { using ::tensorflow::quantization::QuantizationOptions; void AddConvertTpuToCpuModelPasses(mlir::OpPassManager &pm) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 9.9K bytes - Viewed (0)