- Sort Score
- Result 10 results
- Languages All
Results 151 - 160 of 1,398 for TensorT (0.15 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
for (auto arg_num : input_mapping) { const Tensor* tensor; if (auto it = variable_snapshots.find(arg_num); it != variable_snapshots.end()) { tensor = it->second; } else { tensor = inputs[arg_num - num_missing_prefix_ctx_inputs]; } // The input tensor can have the following cases. // 1. Tensor with PjRtTensorBuffer, containing a PjRtBuffer. This case
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/split_merged_operands.cc
// been converted to constants, all the assign ops are gone, etc.). However, // TFLite has these variable tensors semantics. So the variable mapping from TF // to TFLite is actually broken here, we sort of hard-code the variable tensors // based on the actual ops using them, such as unidirectional_sequence_lstm. // // MLIRConverter also benefits from lots of typical compiler optimization like
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
base_offset *= dimension; } return contiguous_offset; } // Performs transposition of a tensor represented as a contiguous element array. // Assumes row-major order. The shape of the input tensor and the desired // permutation is registered during construction, and calling `TransposeValues` // returns the transposed tensor values. class DenseElementsTransposer { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/runtime/runtime.h
// functionality will be supported in the future. // Create a host tensor and copy data into it. // // Raises an error if shape or dtype are incompatible with T. // TODO(b/189458441): Update this when we decide on the representation of // shape and dtype in this API. // Disclaimer: This API is subject to change as we add support for creating // device tensors b/187222691 and enable buffer re-use b/187223179.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/pass_pipeline.h
// required to collect tensor statistics. void AddPreCalibrationPasses( OpPassManager& pm, const ::stablehlo::quantization::CalibrationOptions& calibration_options, const ::stablehlo::quantization::QuantizationSpecs& specs, const ::stablehlo::quantization::DebuggerConfig& debugger_config); // Adds passes for static-range quantization post-calibration. Utilizes tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 12:53:33 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.cc
opt<std::string> input_shapes( "tf-input-shapes", llvm::cl::desc( "Input tensor shapes. Shapes for different tensors are separated by " "':', and dimension sizes for the same tensor are separated by ','"), llvm::cl::init("")); // NOLINTNEXTLINE opt<std::string> output_arrays( "tf-output-arrays", llvm::cl::desc("Output tensor names, separated by ','"), llvm::cl::init("")); // NOLINTNEXTLINE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 10 20:59:50 UTC 2023 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.h
// quantization aware training or calibration, for the remaining tensors. std::vector<std::pair<std::optional<double>, std::optional<double>>> input_ranges; // Whether to disable setting the quantization parameters of the input nodes // using input ranges. bool disable_set_input_nodes_quantization_params = false; // The default ranges can be used when a tensor doesn't have quantization
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 10:16:19 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantization_options.proto
// not applied regardless of the op support. Currently, it is supported for // XLA opset for SRQ on weight tensors (not activation), // and Uniform Quantized opset . optional bool enable_per_channel_quantization = 10; // Enables two inputs of an operation to be both tensors. // Currently supports MatMul and BatchMatMul ops for XLA. // TODO(b/263528090): Check the condition when this feature is beneficial.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 06:31:19 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/jit/variable_info_util.h
// initialized, the value will not be present. using ResourceVarsSnapshot = absl::flat_hash_map<int, std::optional<Tensor>>; // Takes a snapshot of the values of resource variable arguments, whose indices // are specified in `variable_indices` argument. We snapshot tensors that back // resource variables since concurrent updates may modify the shape, and it is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 14 21:57:02 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
} return %graph#0, %graph#1, %graph#2, %graph#3 : !tf_res, !tf_res, tensor<f32>, tensor<f32> } ``` After: ```mlir func @while_body(%arg0: !tf_res, %arg1: !tf_res, %arg2: tensor<f32>, %arg3: tensor<f32>, %chain_0: tensor<i32>, %chain_1: tensor<i32>) -> (!tf_res, !tf_res, tensor<f32>, tensor<f32>, tensor<i32>, tensor<i32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)