Searched refs:kTensor (Results 1 – 15 of 15) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Sparsification.cpp | 54 enum class Kind { kTensor, kInvariant, kMulF, kMulI, kAddF, kAddI }; enumerator
63 assert((kind == Kind::kTensor && e0 != -1u && e1 == -1u && !val) || in TensorExp()
168 if (exp(e).kind == Kind::kTensor && exp(e).e0 == numTensors - 1) in optimize()
370 return merger.addExp(Kind::kTensor, argN); in buildTensorExp()
407 if (kind == Kind::kTensor || kind == Kind::kInvariant) { in buildLattices()
412 unsigned t = kind == Kind::kTensor ? merger.exp(exp).e0 in buildLattices()
420 case Kind::kTensor: in buildLattices()
573 if (merger.exp(exp).kind == Kind::kTensor) in genExp()
580 case Kind::kTensor: in genExp()
598 if (merger.exp(exp).kind == Kind::kTensor) { in genInvariants()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | mangling_util.h | 28 enum class MangledKind { kUnknown, kDataType, kTensorShape, kTensor }; enumerator
|
| D | mangling_util.cc | 57 return MangledKind::kTensor; in GetMangledKind()
|
| D | convert_attr.cc | 49 case AttrValue::kTensor: in ConvertNonFuncAttributeValue()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tensorflow_util.cc | 109 } else if (attr.second.value_case() == AttrValue::kTensor) { in LogDumpGraphDef()
|
| D | import_tensorflow.cc | 132 CHECK_EQ(attr.value_case(), AttrValue::kTensor); in GetTensorAttr()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | attr_value_util.cc | 243 case AttrValue::kTensor: in SummarizeAttrValue()
329 VALIDATE_FIELD(tensor, "tensor", kTensor); in AttrValueHasType()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 62 case AttrValue::kTensor: { in ExtractTensors()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 302 SINGLE_CASE(kTensor, TF_ATTR_TENSOR, -1); in TF_OpKernelConstruction_GetAttrSize()
|
| D | c_api.cc | 1253 SINGLE_CASE(kTensor, TF_ATTR_TENSOR, -1); in TF_OperationGetAttrMetadata()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2046 enum class TrtInputArg { kTensor = 1, kWeight = 2, kBoth = 3 }; enumerator
2071 if (expected_inputs[i].second == TrtInputArg::kTensor && in CheckInputsWeights()
2093 x.first, x.second ? TrtInputArg::kWeight : TrtInputArg::kTensor); in CheckInputsWeights()
4734 params->use_implicit_batch ? TrtInputArg::kTensor : TrtInputArg::kBoth; in ConvertPack()
5114 TrtInputArg expected_input = TrtInputArg::kTensor; in ConvertConcat()
5315 {"indices", TrtInputArg::kTensor}, in ConvertGather()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 228 case AttrValue::kTensor: in PrintAttrValue()
|
| /external/tensorflow/tensorflow/core/summary/ |
| D | summary_db_writer.cc | 1031 case Summary::Value::ValueCase::kTensor: in MigrateSummary()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 1119 case tensorflow::AttrValue::kTensor: in SetOpAttrValueScalar()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 3948 const char kTensor[] = "T"; variable
4108 absl::StrAppend(&result->str, kTensor);
|