- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for _xla (0.44 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
# Check the converted model in the XLA opset. quantization_options = quant_opts_pb2.QuantizationOptions( quantization_method=quant_opts_pb2.QuantizationMethod( preset_method=_PresetMethod.METHOD_STATIC_RANGE_INT8 ), tags=tags, signature_keys=[signature_key], op_set=quant_opts_pb2.XLA, enable_two_input_tensors=not use_kernel, )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
bool ShapeInference::InferShapeForXlaGatherOp(XlaGatherOp op) { xla::Shape input_shape = xla::TypeToShape(op.getOperand().getType()); if (input_shape == xla::Shape() || input_shape.is_unbounded_dynamic()) return false; xla::Shape start_indices_shape = xla::TypeToShape(op.getStartIndices().getType()); if (start_indices_shape == xla::Shape()) return false; xla::GatherDimensionNumbers gather_dim_numbers;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
} def TPUValidateInputsPass : Pass<"tf-tpu-validate-inputs", "ModuleOp"> { let summary = "Validates inputs to the TPU TF/XLA bridge"; let description = [{ This pass checks that the IR has valid input to TPU TF/XLA bridge. It checks the relations of multiple ops. Properties of single ops are checked by the 'verify' method of ops. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
func.func @unparsable_element_input_sharding_configuration(%arg0: tensor<?xi32>) { // expected-error@+1 {{bad 'input_sharding_configuration' attribute at index 0 with value 'test': failed to parse to xla::OpSharding}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
#include "tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" #include "tensorflow/core/framework/kernel_shape_util.h" #include "tensorflow/core/lib/math/math_util.h" #include "tensorflow/core/platform/status.h" #include "tensorflow/core/util/padding.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
DenseIntElementsAttr::get(output_int_type, output_ty.getShape()); // use int32_t instead of int64_t if all elements are in the range of int32 // because int64 is not supported in dynamic reshape in XLA bool elements_all_in_int32_range = std::all_of(output_ty.getShape().begin(), output_ty.getShape().end(), IsWithinInt32Range); if (elements_all_in_int32_range) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
let summary = "DynamicUpdateSlice."; let description = [{ DynamicUpdateSlice op that have the same semantics with XLA DynamicUpdateSlice. Generates a result which is the value of the input array operand, with a slice update overwritten at start_indices. See https://www.tensorflow.org/xla/operation_semantics#dynamicupdateslice. }]; let arguments = (ins
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/error_util.h" #include "tensorflow/compiler/mlir/tensorflow/utils/mangling_util.h" #include "tensorflow/compiler/mlir/tensorflow/utils/translate_utils.h" #include "xla/status_macros.h" #include "tensorflow/core/common_runtime/function.h" #include "tensorflow/core/common_runtime/graph_constructor.h" #include "tensorflow/core/common_runtime/shape_refiner.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)