- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 28 for type_attr (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
"'"); TypeAttr type_attr = mlir::dyn_cast<TypeAttr>(attr); if (!type_attr) return InvalidArgument("Attribute '", output_arg.type_attr(), "' required for output '", output_arg.name(), "' isn't a type attribute"); state_->types.push_back(UnrankedTensorType::get(type_attr.getValue())); } else if (!output_arg.type_list_attr().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype)); AttrValue type_attr; type_attr.set_type(dtype); (*node_def->mutable_attr())["T"] = type_attr; AttrValue index_attr; index_attr.set_i(index); (*node_def->mutable_attr())["index"] = index_attr; if (auto device_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype)); AttrValue type_attr; type_attr.set_type(dtype); (*node_def->mutable_attr())["T"] = type_attr; AttrValue index_attr; index_attr.set_i(index); (*node_def->mutable_attr())["index"] = index_attr; if (auto device_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
} return string(name); } void InferArgAttributes(const OpDef::ArgDef& arg, std::unordered_map<string, string>* inferred_attrs) { if (!arg.type_attr().empty()) { gtl::InsertIfNotPresent(inferred_attrs, arg.type_attr(), arg.name()); } else if (!arg.type_list_attr().empty()) { gtl::InsertIfNotPresent(inferred_attrs, arg.type_list_attr(), arg.name()); } if (!arg.number_attr().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
llvm::SmallVector<Attribute, 2> shape_attrs; 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>(
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/tfr/passes/decompose.cc
} attribute = TypeAttr::get(type); } Value attr_cst; // Wrap these special attributes as a special TFR constant, so the SSA // value has a valid type to be used as TFR function argument. These // attributes are not expected to be manipulated by the lowering passes. if (mlir::isa<TypeAttr>(attribute) || mlir::isa<ArrayAttr>(attribute) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
/*asymmetric_quantize_inputs=*/mlir::BoolAttr(), /*input_to_input_intermediate=*/mlir::TypeAttr(), /*input_to_forget_intermediate=*/mlir::TypeAttr(), /*input_to_cell_intermediate=*/mlir::TypeAttr(), /*input_to_output_intermediate=*/mlir::TypeAttr(), /*effective_hidden_scale_intermediate=*/mlir::TypeAttr()); // Cast the static shaped lstm result to FuncOp's signature -
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/quantization_context.cc
input_specs.push_back(original_input_specs[i]); } else if (requantize.pos == RequantizeState::ON_OUTPUT) { input_specs.push_back(TypeAttr::get(requantize.params)); } else { input_specs.push_back(TypeAttr::get(state.params)); } } op->setAttr("input_specs", ArrayAttr::get(context, input_specs)); llvm::SmallVector<Attribute, 4> output_specs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 01:38:03 UTC 2024 - 13.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
// index. template <typename LstmOp> inline QuantizedType GetIntermediateElementType(LstmOp op, int tensor_index) { if (tensor_index < 0 || tensor_index > 4) return nullptr; TypeAttr attr = op->template getAttrOfType<TypeAttr>( intermediate_attributes[tensor_index]); if (!attr) { return nullptr; } return QuantizedType::getQuantizedElementType(attr.getValue()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
if (input_types.size() <= 1) return; Type target_input_type = mlir::cast<TypeAttr>(input_types[0]).getValue(); auto result_type = UnrankedTensorType::get(target_input_type); for (auto i = 1; i < input_types.size(); ++i) { Type current_input_type = mlir::cast<TypeAttr>(input_types[i]).getValue(); if (current_input_type != target_input_type) { input_values[i] =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0)