- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 46 for TypeAttr (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
builder.create<tf_saved_model::GlobalTensorOp>( NameLoc::get(builder.getStringAttr(name.str())), builder.getStringAttr(name), tensor_attr, TypeAttr::get(tensor_attr.getType()), builder.getUnitAttr()); } return success(); } } // namespace LogicalResult LiftVariables(ModuleOp module, Session* session) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_ops.td
sense of `tf.TensorShape` compatibility. And the element types must match. }]; let arguments = (ins StrAttr:$sym_name, OptionalAttr<ElementsAttr>:$value, TypeAttr:$type, UnitAttr:$is_mutable ); let hasVerifier = 1; } def TfSavedModel_SessionInitializerOp: TfSavedModel_Op<"session_initializer"> { let summary = "Initializes TensorFlow session state.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
// determined while going through quantization passes. OptionalAttr<TypeAttr>:$input_to_input_intermediate, OptionalAttr<TypeAttr>:$input_to_forget_intermediate, OptionalAttr<TypeAttr>:$input_to_cell_intermediate, OptionalAttr<TypeAttr>:$input_to_output_intermediate, OptionalAttr<TypeAttr>:$effective_hidden_scale_intermediate ); let results = (outs AnyTensor:$output);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
"mlir::OperandElementTypeIterator(values.end())};", [{ ArrayAttr::get($_ctxt, [&]() { llvm::SmallVector<Attribute, 4> ret; for (auto t : $_self) ret.push_back(TypeAttr::get(t)); return ret; }()) }] >; // A derived attribute that returns the shapes of the tensors in the actual // value pack that corresponds to the `idx`-th ODS-declared variadic operand.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
llvm::SmallVector<Attribute, 2> type_attrs; for (Type type : dataset_types) { shape_attrs.push_back( TF::ShapeAttr::get(builder.getContext(), mlir::cast<ShapedType>(type))); type_attrs.push_back(TypeAttr::get(getElementTypeOrSelf(type))); } auto anonymous_iterator = builder.create<AnonymousIteratorV3Op>( reduce_dataset.getLoc(), RankedTensorType::get({}, builder.getType<ResourceType>()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
// Here TArgs types do not include types of the first two parameters, // i.e. the convolution input and the filter. TArgs are parameters for // the extras like the bias etc. auto attr = TypeAttr::get(getElementTypeOrSelf(contraction.getType())); SmallVector<Attribute, 4> targs_values(operands.size() - 2, attr); ArrayAttr targs_attr = ArrayAttr::get(context, targs_values); attrs.push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
res_types, res_attrs))) return failure(); // Update the function type which has excluded the control args. func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType( arg_types, res_types))); // Update arg/result attributes. func.setAllArgAttrs(arg_attrs); func.setAllResultAttrs(res_attrs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
auto min_max = GetMinMaxValuesForArgument(func_name, i); // The input min/max or mean/std are not specified, then skip. if (!min_max.first.has_value() || !min_max.second.has_value()) return; TypeAttr params = quant::GetQuantizedTypeAttr( builder, input_type, builder.getF64FloatAttr(min_max.first.value()), builder.getF64FloatAttr(min_max.second.value()),
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.cc
auto min_max = GetMinMaxValuesForArgument(func_name, i); // The input min/max or mean/std are not specified, then skip. if (!min_max.first.has_value() || !min_max.second.has_value()) return; TypeAttr params = quant::GetQuantizedTypeAttr( builder, input_type, builder.getF64FloatAttr(min_max.first.value()), builder.getF64FloatAttr(min_max.second.value()),
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/lite/experimental/tac/transforms/device_transform_patterns.cc
mean_op.getAxis(), mean_op.getKeepDims()); // Insert a requant op. rewriter.replaceOpWithNewOp<TFL::QuantizeOp>( mean_op, output_type, new_mean_op, mlir::TypeAttr::get(output_type)); return success(); } } // namespace tac } // namespace TFL
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0)