- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 380 for getLoc (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_canonicalization_helper.h
// another TF op. // TODO(jpienaar): This is a workaround until behavior is established. template <typename OpTy, typename... Args> OpTy CreateTfOp(RewriterBase &b, Operation *op, Args &&...args) { auto ret = b.create<OpTy>(op->getLoc(), std::forward<Args>(args)...); CopyDeviceAndUnderscoredAttributes(op, ret.getOperation()); return ret; } // Helper function to replace TF op with another op while copying all underscore
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 12 21:57:12 UTC 2021 - 2.4K bytes - Viewed (0) -
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/python/mlir_wrapper/basic_classes.cc
py::return_value_policy::reference) .def("end", &mlir::Block::end) .def("addArgument", [](mlir::Block& block, mlir::Type type) { return block.addArgument(type, block.getParent()->getLoc()); }); py::class_<mlir::Value>(m, "Value").def("getType", &mlir::Value::getType); py::class_<mlir::OpResult, mlir::Value>(m, "OpResult"); py::class_<mlir::BlockArgument, mlir::Value>(m, "BlockArgument");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 22 04:26:07 UTC 2022 - 2.4K 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)