Home
last modified time | relevance | path

Searched defs:int_attr (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dnms_utils.cc140 auto int_attr = attrs.get(attribute).dyn_cast_or_null<IntegerAttr>(); in AddIntAttr() local
163 auto int_attr = attrs.get(attribute).dyn_cast_or_null<IntegerAttr>(); in HasIntAttr() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Drewrite_util.h60 DenseIntElementsAttr int_attr; in IsConstantValueOf() local
Deinsum.cc62 auto int_attr = IntegerAttr::get(rewriter->getIntegerType(32), value); in createI32ConstOp() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_legalize_i1_type.cc95 if (auto int_attr = attr.getValue().dyn_cast<DenseIntElementsAttr>()) { in isLegalAttribute() local
109 if (auto int_attr = attr.getValue().dyn_cast<DenseIntElementsAttr>()) { in convertAttribute() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dfill_spmd_expander.cc67 auto int_attr = mlir::DenseIntElementsAttr::get(int_type, shard_values); in ExpandOp() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc85 } else if (auto int_attr = attr.second.dyn_cast_or_null<IntegerAttr>()) { in CreateTflFusableOpCustomOptions() local
/external/pytorch/test/cpp/jit/
Dtest_module_api.cpp192 auto int_attr = "int_attr"; in TEST() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc2313 auto int_attr = attr.cast<DenseIntElementsAttr>(); in fold() local
Dtf_ops_n_z.cc2266 auto int_attr = entry.value().cast<IntegerAttr>(); in verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc690 auto int_attr = input.cast<DenseIntElementsAttr>(); in SliceDenseIntElementsAttrColumn2D() local
4021 auto int_attr = attr.cast<DenseIntElementsAttr>(); in ConvertAxisAttr() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2327 IntegerAttr int_attr = *attr.getValues<IntegerAttr>().begin(); in ExtractConstantIntFromTensor() local