- Sort Score
- Result 10 results
- Languages All
Results 221 - 230 of 258 for created (0.08 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
llvm::StringRef device) { builder->setInsertionPoint(op); auto launch = builder->create<tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new Block); op->replaceAllUsesWith(launch); builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc
// the resource, and we use it to just look up. // See core/kernels/data/iterator_ops.cc.) resource_op = builder.create<TF::IteratorOp>( op->getLoc(), resource.getType(), shared_name, container, it.getOutputTypes(), it.getOutputShapes()); } else { resource_op = builder.create<TF::VarHandleOp>( op->getLoc(), resource.getType(), container, shared_name); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_xla_call_module_op_to_bfloat16.cc
builder.setInsertionPoint(op); for (auto& op_operand : op->getOpOperands()) { if (IsLargeFloatType(op_operand.get().getType())) { op_operand.set(builder.create<TF::CastOp>( op->getLoc(), ToBfloat16Type(op_operand.get().getType()), op_operand.get())); } } builder.setInsertionPointAfter(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
} // Rewrite return if there are variable writes. const int return_operands_size = return_operands.size(); if (return_operands_size > num_results_before) { builder.create<func::ReturnOp>(return_op.getLoc(), return_operands); return_op.erase(); } // Update function argument and result types with new resource subtypes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
Value value = effect.getValue(); found_any_effect = true; // We only collect value-based side effects here for which we can use // resource alias analysis. Other side effects are treated as op-based // side effects. if (!ShouldUseResourceAliasAnalysis(effect)) continue; if (value) processed_values.insert(value); TypeID type_id = effect.getResource()->getResourceID();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/converter_gen.cc
assert(op_name.starts_with("TFL_") && "unexpected op prefix"); assert(op_name.ends_with("Op") && "unexpected op suffix"); // E.g., AddOp -> CreateAddOperator std::ostringstream oss; oss << "Create" << op_name.drop_front(4).str() << "erator"; return oss.str(); } static inline bool IsLstmOp(const StringRef op_name) { return op_name.take_back(6) == "LSTMOp"; } static int HasOptions(const Record &def) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 15:05:28 UTC 2023 - 23.7K bytes - Viewed (0) -
tensorflow/cc/framework/scope.cc
} CompositeOpScopes Scope::GetCompositeOpScopes( const string& composite_op_name) const { if (impl()->op_name_.empty() && composite_op_name.empty()) { UpdateStatus(errors::InvalidArgument( "Cannot create composite op scopes with empty name")); return {*this, *this}; } if (!impl()->single_use_scope()) { Scope child = NewSubScope(impl()->op_name_.empty() ? composite_op_name
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc
builder.setInsertionPoint(composite); if (IsKVCacheCompositeOp(composite)) { return builder.create<TFL::CustomOp>( composite->getLoc(), composite->getResultTypes(), composite->getOperands().slice(2, 3), composite.getName(), CustomOption(&builder, custom_option_buffer)); } return builder.create<TFL::CustomOp>( composite->getLoc(), composite->getResultTypes(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
if (can_eliminate(result)) continue; new_result_types.push_back(result.getType()); } // Rebuild the new operation with lesser number of results. OpBuilder builder(op); Operation *new_op = Operation::create( op->getLoc(), op->getName(), new_result_types, op->getOperands(), op->getAttrs(), op->getPropertiesStorage(), op->getSuccessors(), op->getNumRegions()); builder.insert(new_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/custom_call.cc
namespace mlir { namespace odml { LogicalResult ConvertCustomCallOp::matchAndRewrite( mhlo::CustomCallOp mhlo_custom_call, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const { auto tfl_custom = rewriter.create<TFL::CustomOp>( mhlo_custom_call.getLoc(), mhlo_custom_call.getResultTypes(), mhlo_custom_call.getInputs()); tfl_custom.setCustomCodeAttr( rewriter.getStringAttr(mhlo_custom_call.getCallTargetName()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.3K bytes - Viewed (0)