- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 144 for _xla (0.29 sec)
-
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/jit/resource_operation_safety_analysis.cc
// resource op kind `from` to a TensorFlow operation of resource op kind `to` // can be represented by an XLA cluster and needs no special handling around // auto-jit. bool IsEdgeSafe(XlaResourceOpKind from, XlaResourceOpKind to) { // XLA clusters force all reads to happen before all writes. Moreover the set // of reads are executed as one atomic operation, and the set of writes are as
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 11.2K 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/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/jit/rearrange_function_argument_pass_test.cc
#include "tensorflow/cc/ops/functional_ops.h" #include "tensorflow/cc/ops/standard_ops.h" #include "tensorflow/compiler/jit/encapsulate_util.h" #include "tensorflow/compiler/tf2xla/rearrange_function_argument.h" #include "xla/test.h" #include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/function.h" #include "tensorflow/core/framework/common_shape_fns.h" #include "tensorflow/core/framework/function.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
// similar manner. // // This pass may be used to avoid placing shape-determining constants in the CPU // graph and pass them as arguments to the TPU graph (via `TPUPartitionedCall`). // If this happens, the XLA compiler cannot recognize such arguments as // constants and may result in an error. // // A set of predefined ops and operand indices is used to determine whether an // operand is a target for constant duplication.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
"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"))}; }; bool QuantizePass::shouldKeepUnusedQdqPattern() { return target_opset_ == OpSet::XLA && (quant_specs_.weight_only_quantization || quant_specs_.weight_quantization);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir
func.return } // CHECK-LABEL: func @empty_func() { func.func @empty_func() { func.return } // ----- // Tests with a block argument inputs/outputs with no xla sharding op attached // gets default maximal(0) sharding configuration. // CHECK-LABEL: func @check_default_sharding_for_block_arg_inputs_outputs func.func @check_default_sharding_for_block_arg_inputs_outputs(%arg0: tensor<*xi32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 20 19:07:52 UTC 2024 - 47.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.cc
#include "tensorflow/compiler/mlir/tosa/tf_tfl_passes.h" #include "tensorflow/compiler/mlir/tosa/tfl_passes.h" #include "tensorflow/compiler/mlir/tosa/transforms/passes.h" #include "xla/mlir/framework/transforms/passes.h" #include "xla/mlir_hlo/mhlo/transforms/passes.h" #include "xla/status_macros.h" #include "tensorflow/core/common_runtime/eager/context.h" #include "tensorflow/core/common_runtime/function_body.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0)