- Sort Score
- Result 10 results
- Languages All
Results 191 - 200 of 1,825 for Auto (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.cc
// Removes relevant ops in topological order. for (auto& op : vars) RecursiveRemove(op, erase_list, dead_ops); // Erases the ops. for (auto op : erase_list) op->erase(); } void RemoveVariablesInSessionInitializerPass::runOnOperation() { ModuleOp module_op = getOperation(); for (auto init_func_op : GetInitializerFunctions(module_op)) { if (!init_func_op) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc
if (options.has_chain) result.types.push_back(chain_type); result.types.append(num_results, tensor_type); llvm::SmallVector<mlir::Attribute, 4> op_attr_array; for (const auto &key_value : op_attrs) { auto key = key_value.getName(); auto value = key_value.getValue(); op_attr_array.push_back(builder.getArrayAttr({key, value})); } result.attributes.push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
SmallVectorImpl<Type>* output_types) const { for (auto res : llvm::enumerate(signature.getResults())) { if (auto tensor_type = mlir::dyn_cast<TFRTensorType>(res.value())) { // tfr.tensor should only have one attribute attached. auto attr_key = tensor_type.getAttrKeys().front(); Builder builder(signature.getContext()); if (auto attr = attrs.lookup(attr_key.getValue())) { output_types->push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
StringRef func_name = func.getName(); auto has_quantize_op = [&](const Value arg) { return (arg.hasOneUse() && llvm::isa<quantfork::QuantizeCastOp>(*arg.user_begin())); }; bool need_to_set_input_nodes_quantization_params = false; for (const BlockArgument arg : func.getArguments()) { auto shaped = mlir::dyn_cast<ShapedType>(arg.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
bool metadata_updated = false; auto maybe_replicate = execute_launches.front()->getParentOfType<tf_device::ReplicateOp>(); for (auto execute_and_input_mapping : llvm::zip(execute_launches, input_mappings)) { auto& execute_launch = std::get<0>(execute_and_input_mapping); auto execute = llvm::cast<TF::TPUExecuteOp>(execute_launch.GetBody().front());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize_global_tensors.cc
GlobalTensorUsesMap global_tensor_uses; SymbolTable symbol_table(module); for (auto func : module.getOps<func::FuncOp>()) { for (size_t i = 0, e = func.getNumArguments(); i < e; i++) { auto sym = func.getArgAttrOfType<SymbolRefAttr>(i, "tf_saved_model.bound_input"); if (!sym) { continue; } auto global_tensor = symbol_table.lookup<GlobalTensorOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
Value resource; if (auto read = llvm::dyn_cast<TF::ReadVariableOp>(op)) { resource = read.getResource(); } else if (auto write = llvm::dyn_cast<TF::AssignVariableOp>(op)) { resource = write.getResource(); } const TF::ResourceDataflowState* state = solver.lookupState<TF::ResourceDataflowState>(resource); if (!state) { return nullptr; } auto ops = state->getValue().ops;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
} // Make a copy of the quantized function. auto module = call_op->getParentOfType<ModuleOp>(); SymbolTable symbol_table(module); const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(call_op.getFAttr()); const auto float_func = dyn_cast<func::FuncOp>(symbol_table.lookup(f_attr.getValue())); rewriter.setInsertionPointAfter(float_func);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
StringAttr::get(input_defining_op->getContext(), "auto")); outside_compiled_ops.push(input_defining_op); } } // Mark for outside compilation any operations that consume variant // outputs from an outside compiled operation. for (auto value : op->getResults()) { if (IsVariant(value)) { for (auto user : value.getUsers()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0)