Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dconvert_type.cc147 mlir::Type GetShapeStrippedType(mlir::TypeAttr type_attr) { in GetShapeStrippedType()
/external/tensorflow/tensorflow/core/framework/
Dop_def.proto40 string type_attr = 4; // if specified, attr must have type "type" field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc371 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create() local
395 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create() local
414 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc109 DataType GetDataType(const NodeDef& node, const TypeAttrId& type_attr) { in GetDataType()
130 bool SetDataType(NodeDef* node, const TypeAttrId& type_attr, DataType type) { in SetDataType()
302 TypeAttrId type_attr = GetTypeAttrId(arg_def, arg_inds.second); in AddNode() local
317 TypeAttrId type_attr = GetTypeAttrId(arg_def, arg_inds.second); in AddNode() local
370 TypeAttrId type_attr; member
499 for (const TypeAttrId& type_attr : node_type_map.GetTypeAttrs(node)) { in InitializeFromGraph() local
1413 for (const TypeAttrId& type_attr : node_type_map_.GetTypeAttrs(node)) { in GetTensorListFloat32NodeTypeId() local
1898 for (const TypeAttrId& type_attr : node_type_map_.GetTypeAttrs(*node)) { in ChangeTypeAttrsAndAddCasts() local
Dremapper.cc215 const string& type_attr = "T") { in HaveSameDataType()
224 const string& type_attr = "T") { in HasDataType()
230 const string& type_attr = "T") { in IsCpuCompatibleDataType()
254 const string& type_attr = "T") { in IsCpuCompatibleDataType()
Dconstant_folding.cc3874 const DataType type_attr = node->attr().at("dtype").type(); in AddQuantizedMatMulMinMaxOutConstNodes() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddefault_quant_params.cc180 TypeAttr type_attr = TypeAttr::get(new_type); in QuantizeValue() local
Doptimize.cc250 static Type GetShapeStrippedType(TypeAttr type_attr) { in GetShapeStrippedType()
/external/tensorflow/tensorflow/python/framework/
Dpython_api_info.h95 Attribute* type_attr; // not owned. member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc230 AttrValue type_attr; in GetArgumentNode() local
266 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.cc367 const std::string& type_attr = input_def.type_attr(); in MaybeInferSingleInputAttrs() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc312 DataType GetDataTypeFromAttr(const NodeDef& node, const string& type_attr) { in GetDataTypeFromAttr()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc617 mlir::TypeAttr type_attr = in AddOpIntermediatesForLstm() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc694 auto type_attr = op.compare_typeAttr(); in ExportXlaOp() local
/external/libxml2/
DHTMLparser.c762 static const char* const type_attr[] = { "type", NULL } ; variable
/external/python/cpython3/Modules/_ctypes/
D_ctypes.c1498 PyObject *length_attr, *type_attr; in PyCArrayType_new() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1062 StringAttr type_attr; in BuildSortComparisonBody() local