Home
last modified time | relevance | path

Searched defs:type_attr (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/
Dpython_test_attrs.cc45 auto type_attr = attribute.getValue().dyn_cast<TypeAttr>(); in verifyRegionArgAttribute() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dconvert_type.cc192 mlir::Type GetShapeStrippedType(mlir::TypeAttr type_attr) { in GetShapeStrippedType()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc172 DataType GetDataType(const NodeDef& node, const TypeAttrId& type_attr) { in GetDataType()
193 bool SetDataType(NodeDef* node, const TypeAttrId& type_attr, DataType type) { in SetDataType()
368 TypeAttrId type_attr = GetTypeAttrId(arg_def, arg_inds.second); in AddNode() local
383 TypeAttrId type_attr = GetTypeAttrId(arg_def, arg_inds.second); in AddNode() local
436 TypeAttrId type_attr; member
565 for (const TypeAttrId& type_attr : node_type_map.GetTypeAttrs(node)) { in InitializeFromGraph() local
1326 for (const TypeAttrId& type_attr : in IsQuantized() local
1556 for (const TypeAttrId& type_attr : node_type_map_.GetTypeAttrs(node)) { in GetTensorListFloat32NodeTypeId() local
2147 const TypeAttrId& type_attr, NodeDef* node, in CollectOutputPorts()
2173 for (const TypeAttrId& type_attr : node_type_map_.GetTypeAttrs(*node)) { in ChangeTypeAttrsAndAddCasts() local
Dremapper.cc273 const string& type_attr = "T") { in HaveSameDataType()
282 const string& type_attr = "T") { in HasDataType()
288 const string& type_attr = "T") { in IsCpuCompatibleDataType()
309 const string& type_attr = "T") { in IsGpuCompatibleDataType()
/external/tensorflow/tensorflow/core/framework/
Dop_def.proto43 string type_attr = 4; // if specified, attr must have type "type" field
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dcorert_converter.cc303 if (auto type_attr = attr.dyn_cast<mlir::TypeAttr>()) { in ConvertAttribute() local
354 mlir::TypeAttr CoreRTConverter::ConvertTypeAttribute(mlir::TypeAttr type_attr) { in ConvertTypeAttribute()
Dtf_to_tfrt_data.cc56 if (T type_attr = value_attr.dyn_cast<T>()) { in ConstAttrToTypeAttr() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc372 TypedAttr type_attr = attr.dyn_cast<TypedAttr>(); in Create() local
396 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create() local
415 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_batch_kernels.cc108 auto type_attr = attr.cast<tfrt::TypeAttr>(); in ExecuteFallbackOp() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dmodify_io_nodes.cc177 TypeAttr type_attr = TypeAttr::get(returned_type); in ModifyOutputNodes() local
Ddefault_quant_params.cc187 TypeAttr type_attr = TypeAttr::get(new_type); in QuantizeValue() local
Doptimize.cc314 static Type GetShapeStrippedType(TypeAttr type_attr) { in GetShapeStrippedType()
/external/tensorflow/tensorflow/core/ir/importexport/
Dfunctiondef_export.cc118 if (StringAttr type_attr = arg_attrs.getAs<StringAttr>("tfg.type_attr")) in ExportArgDef() local
/external/tensorflow/tensorflow/python/framework/
Dpython_api_info.h95 Attribute* type_attr; // not owned. member
Dop_def_library_pybind.cc324 const std::string& type_attr, const AllowedAttrMap& allowed_list_attr_map, in AssertSatisfiesAllowedListAttrTypeConstraint()
457 const std::string& type_attr = input_arg.type_attr(); in GetInputType() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc210 AttrValue type_attr; in GetArgumentNode() local
246 AttrValue type_attr; in GetReturnNode() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc274 for (const string& type_attr : type_attrs) { in RegisterCompilationKernels() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc369 const std::string& type_attr = input_def.type_attr(); in MaybeInferSingleInputAttrs() local
/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dattr_util.cc559 } else if (auto type_attr = bef_attr.dyn_cast<tfrt::TypeAttr>()) { in SetUpScalarAttr() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc297 DataType GetDataTypeFromAttr(const NodeDef& node, const string& type_attr) { in GetDataTypeFromAttr()
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc1055 auto type_attr = op->getAttrOfType<TypeAttr>("T"); in matchAndRewrite() local
2224 TypeAttr type_attr = op->getAttrOfType<TypeAttr>("T"); in matchAndRewrite() local
2349 TypeAttr type_attr = op->getAttrOfType<TypeAttr>("T"); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc781 mlir::TypeAttr type_attr = in AddOpIntermediatesForLstm() local
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc1968 const std::string& type_attr = input_def.type_attr(); in MaybeInferInputAttrs() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc403 Type GetType(Attribute shape_attr, Attribute type_attr) { in GetType()
/external/cronet/stable/third_party/libxml/src/
DHTMLparser.c663 static const char* const type_attr[] = { "type", NULL } ; variable

12