- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 168 for setFAttr (0.11 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
class ConvertEmbeddedLookupFunc { public: explicit ConvertEmbeddedLookupFunc(func::FuncOp func) : func_(func) {} void RewriteFunc() { func_->setAttr(kTFImplements, StringAttr::get(func_.getContext(), "embedding_lookup")); Value lookup = func_.getArgument(1); Value value = func_.getArgument(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
if (!preceding_qcast) return failure(); auto new_qcast = rewriter.create<quantfork::QuantizeCastOp>( q_op.getLoc(), q_op.getType(), preceding_qcast.getArg()); new_qcast->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); q_op->replaceAllUsesWith(new_qcast); return success(); } }; bool PrepareQuantizePass::ContainsQuantizeOps(func::FuncOp func) {
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/lite/transforms/prepare_quantize_dynamic_range.cc
// tensorflow/lite/kernels/fully_connected.cc. The kernels will handle // the asymmetric_quantize_inputs attribute in the builtin option. dynamic_range_quantized_user->setAttr( kAsymmetricQuantizeInputsAttr, BoolAttr::get(rewriter.getContext(), true)); changed = true; } } return changed; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
xla::OpSharding sharding = manual_sharding ? ::xla::sharding_builder::Manual() : ::xla::sharding_builder::AssignDevice(kShardingTpuCore); op->setAttr(kShardingAttr, StringAttr::get(op->getContext(), sharding.SerializeAsString())); } // Assigns frontend attributes holding information about data type and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
to_keep.push_back(sym_ref); } } for (auto func_op : to_remove) rewriter.eraseOp(func_op); if (to_keep.empty()) rewriter.eraseOp(op); else op->setAttr("initializers", rewriter.getArrayAttr(to_keep)); return success(); } }; void SessionInitializerOp::getCanonicalizationPatterns( RewritePatternSet &results, MLIRContext *context) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
for (NamedAttribute attr : op->getAttrs()) { auto symref = mlir::dyn_cast<FlatSymbolRefAttr>(attr.getValue()); if (!symref) continue; if (symref.getValue() != func.getName()) continue; op->setAttr(attr.getName(), FlatSymbolRefAttr::get(op->getContext(), cloned.getName())); break; } } // Traverse results backward so that indices to be deleted stay unchanged.
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/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
.getInt(); arg_metadata->set_retval_index_for_sharding(ret_index); } } // Update the metadata of the compile op. compile.setAttr("metadata", StringAttr::get(compile.getContext(), metadata.SerializeAsString())); return mapping; } // Adds a new replicated input to the replicate op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
updated_input_shapes_attr.emplace_back(input_shapes_attr[i]); } // Replaces the attribute with the updated "#tf_type.shape" array. // Builder builder(func_op.getContext()); func_op->setAttr( kTfInputShapesAttr, ArrayAttr::get(func_op.getContext(), updated_input_shapes_attr)); } } // Validates func ops. Returns `failure` if the function is invalid.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_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())); for (int64_t i = 0; i < call.getNumResults(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
key_and_value_pair[0], "' attribute")); continue; } inner_op.setAttr(llvm::StringRef(std::string(key_and_value_pair[0])), identifier_to_attr[llvm::StringRef( std::string(key_and_value_pair[1]))]); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0)