- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 363 for getLoc (0.4 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc
tensorflow::Algorithm tensorflow_rng = tensorflow::ToTensorflowAlgorithm(xla_rng); OpBuilder opbuilder(get_alg_op); auto tf_const = opbuilder.create<TF::ConstOp>( get_alg_op->getLoc(), opbuilder.getI32IntegerAttr((int)tensorflow_rng)); get_alg_op->replaceAllUsesWith(tf_const); get_alg_op->erase(); return success(); } void TFXLADeviceSpecificTransforms::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
if (existing) return *existing; } ModuleOp module = region.getParentOfType<ModuleOp>(); OpBuilder builder(module.getContext()); auto loc = region.getParentOp()->getLoc(); Block& entry = region.front(); int num_region_arguments = entry.getNumArguments(); Operation* terminator = entry.getTerminator(); // Build the function type. Region arguments and extern values together
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
for (Value result : subgraph.FuncOutputs()) { Value cloned_output = values_in_scope.lookup(result); return_operands.push_back(cloned_output); } function_builder.create<mlir::func::ReturnOp>(new_func.getLoc(), return_operands); ops_added.func_op = new_func; module.push_back(new_func); return new_func; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
builder_.setInsertionPointAfter(op); const Value original_result = op->getResult(result_index); QuantizeValue(original_result, quantized_type, op->getLoc()); } void QuantizationDriver::QuantizeArg(BlockArgument arg, const QuantizedType quantized_type) { builder_.setInsertionPointToStart(arg.getOwner());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
indices_type = RankedTensorType::get(new_start_indices_shape, indices_type.getElementType()); indices = rewriter.create<mhlo::ReshapeOp>(parent_op->getLoc(), indices_type, indices); } else if (index_vector_dim != indices_type.getRank() - 1) { // If index_vector_dim isn't the last dimension in indices then it isn't
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc
std::unique_ptr<llvm::MemoryBuffer> mem_buffer; llvm::StringRef quantized_function_library = GetFunctionLibrary(quantization_method_, op_set_); if (quantized_function_library.empty()) { emitError(module.getLoc()) << "Failed to get function library for the opset."; signalPassFailure(); return; } mem_buffer = llvm::MemoryBuffer::getMemBuffer(quantized_function_library,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
// Create the TF op OperationState op_state(op.getLoc(), tf_op_full_name); op_state.addOperands(op.getOperands()); op_state.addTypes(op.getResultTypes()); SmallVector<NamedAttribute, 2> attrs; std::string parsed_op_name; tensorflow::NodeDef node_def; if (failed(ParseCustomOption(op.getCustomOption().getValue(), op.getLoc(), parsed_op_name, attrs, node_def))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
public: LogicalResult matchAndRewrite(ConstantTensorOp cst_tensor_op, PatternRewriter &rewriter) const override { Location loc = cst_tensor_op.getLoc(); Type out_type = cst_tensor_op.getType(); Operation *new_cst = nullptr; ArrayAttr array; if (matchPattern(cst_tensor_op.getArg(), m_Constant(&array))) { llvm::DenseSet<Type> all_types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
contraction, "cannot fuse with the subsequent op as it's not supported by the " "target device."); } SmallVector<Location, 3> locations{contraction.getLoc(), bias_add.getLoc()}; SmallVector<Attribute, 2> fused_ops{StringAttr::get( context, bias_add.getOperation()->getName().stripDialect())}; // BiasAdd may or may not feed into an activation function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0)