- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 53 for get_attr (0.12 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
if (!entry_function_attr) return std::nullopt; return entry_function_attr.getValue(); } else { TF::PartitionedCallOp call_op = dyn_cast_or_null<TF::PartitionedCallOp>(op); const auto f_attr = call_op.getFAttr().dyn_cast<FlatSymbolRefAttr>(); if (!f_attr) return std::nullopt; return f_attr.getValue(); } } class InsertCustomAggregationOpsPass : public PassWrapper<InsertCustomAggregationOpsPass,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
added_func_op->setAttr(kInterfaceNameAttr, interface_name); added_call_op->setAttr(kInterfaceNameAttr, interface_name); StringAttr device = mlir::cast<StringAttr>( added_func_op->getRegion(0).getBlocks().front().front().getAttr(kDevice)); StringAttr inference_type = mlir::cast<StringAttr>( added_func_op->getRegion(0).getBlocks().front().front().getAttr( kInferenceType));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
BlockArgument block_arg) { assert(block_arg.getOwner() == &GetBody()); unsigned arg_number = block_arg.getArgNumber(); unsigned num_replicated_args = GetNumReplicatedBlockArguments(); int32_t num_replicas = getNAttr().getInt(); MutableArrayRef<OpOperand> operands = getOperation()->getOpOperands(); // All replicated arguments are before packed arguments so return replicated
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
auto *new_ewise_op = rewriter.create(op->getLoc(), op->getName().getIdentifier(), {tpose_arg1.getOperand(0), tpose_arg2.getOperand(0)}, new_out_type, op->getAttrs()); // Apply original tranpose to output of ewise op. auto out_tpose_op = rewriter.create<TFL::TransposeOp>( new_ewise_op->getLoc(), op->getResult(0).getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
fastapi/openapi/utils.py
if operation_id in operation_ids: message = ( f"Duplicate Operation ID {operation_id} for function " + f"{route.endpoint.__name__}" ) file_name = getattr(route.endpoint, "__globals__", {}).get("__file__") if file_name: message += f" at {file_name}" warnings.warn(message, stacklevel=1) operation_ids.add(operation_id)
Registered: Mon Jun 17 08:32:26 UTC 2024 - Last Modified: Thu Apr 18 19:40:57 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
} OpBuilder builder(call); auto new_call = builder.create<CallOp>( call.getLoc(), info.decomposed_callee.getFunctionType().getResults(), new_operands, call->getAttrs()); new_call->setAttr( "f", SymbolRefAttr::get( builder.getContext(), const_cast<func::FuncOp&>(info.decomposed_callee).getName()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
EXPECT_TRUE(isa<TF::XlaCallModuleOp>(lifted_op)); EXPECT_EQ( mlir::cast<StringAttr>(lifted_op->getAttr("_original_entry_function")), "composite_dot_general_fn_1"); EXPECT_EQ( mlir::cast<ArrayAttr>(lifted_dot_general_op->getAttr("precision_config")), builder_.getArrayAttr(SmallVector<Attribute>( 1, mlir::stablehlo::PrecisionAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
if (auto device = res->getAttrOfType<StringAttr>("device")) { if (!device.getValue().empty()) result_id->setAttr("device", device); else result_id->setAttr("device", compile_device_op); } else if (compile_device_op) { result_id->setAttr("device", compile_device_op); } res.getOutput().replaceAllUsesWith(compile_op->getResult(0)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
for (const auto& attr_def : op_reg_data->op_def.attr()) { registered_attrs.insert(attr_def.name()); } // Attributes are not in the registered attributes set will be ignored. for (auto& attr : inst->getAttrs()) { if (registered_attrs.find(attr.getName()) == registered_attrs.end()) { attrs_to_ignore->insert( absl::string_view(attr.getName().data(), attr.getName().size())); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
auto range = llvm::make_filter_range(op->getUsers(), [&](Operation *user) { bool same_block = user->getBlock() == op->getBlock(); bool same_device = op->getAttr(kDeviceAttr) == user->getAttr(kDeviceAttr); return same_block && same_device && state.IsMember(user); }); return {range.begin(), range.end()}; } // Users of block argument must be in the same block.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0)