| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_shape_or_rank.cc | 28 if (!(op->type == OperatorType::kShape || op->type == OperatorType::kRank)) { in Run()
53 if (op->type == OperatorType::kShape) { in Run()
|
| D | propagate_array_data_types.cc | 78 case OperatorType::kShape: { in Run()
|
| D | quantize.cc | 93 OperatorType::kShape, in SupportsQuantization()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | scatter_nd.cc | 32 constexpr int kShape = 2; variable
82 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kShape, &shape)); in Prepare()
176 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kShape, &shape)); in Eval()
|
| /external/gemmlowp/public/ |
| D | map.h | 87 static constexpr VectorShape kShape = tShape; variable
116 static constexpr VectorShape kShape = tShape; variable
|
| /external/skia/experimental/skrive/include/ |
| D | SkRive.h | 81 kShape, enumerator
259 Shape() : INHERITED(Type::kShape) {} in Shape()
280 if (t == Type::kShape ) return std::is_base_of<T, Shape >::value; in is_base_of()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_attr.cc | 47 case AttrValue::kShape: in ConvertNonFuncAttributeValue()
|
| D | export_utils.cc | 201 case AttrValue::kShape: in ConvertAttribute()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Vectorization.cpp | 544 ArrayRef<int64_t> kShape = kShapeType.getShape(); in matchAndRewrite() local
553 if (!mask[i] && (inShape[i] != 1 || kShape[i] != 1)) in matchAndRewrite()
556 if (mask[i] && inShape[i] != kShape[i]) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tensorflow_util.cc | 102 } else if (attr.second.value_case() == AttrValue::kShape) { in LogDumpGraphDef()
|
| D | model.h | 127 kShape, enumerator
1397 TensorFlowShapeOperator() : Operator(OperatorType::kShape) {}
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | attr_value_util.cc | 241 case AttrValue::kShape: in SummarizeAttrValue()
328 VALIDATE_FIELD(shape, "shape", kShape); in AttrValueHasType()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 180 {{OperatorType::kShape, 1}, "1.10.0"}, in GetMinimumRuntimeVersionForModel()
|
| D | operator_test.cc | 545 SerializeAndDeserialize(GetOperator("SHAPE", OperatorType::kShape), op); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | codegen_test_h.golden | 402 static const ::xla::ProgramShapeProto* kShape = []() {
407 return kShape;
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_def_util.cc | 351 case tensorflow::AttrValue::kShape: in AttrValueToPyObject()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 300 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OpKernelConstruction_GetAttrSize()
|
| D | c_api.cc | 1251 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OperationGetAttrMetadata()
1460 if (attr->value_case() != tensorflow::AttrValue::kShape) { in TF_OperationGetAttrTensorShapeProto()
|
| /external/gemmlowp/internal/ |
| D | output.h | 324 const int pos = VectorType::kShape == VectorShape::Row ? col : row;
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 226 case AttrValue::kShape: in PrintAttrValue()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 946 const Tensor kShape = test::AsTensor<int32>({32, 32}); in TEST_F() local
955 {{"shape"}, "Const", {}, {{"value", kShape}, {"dtype", DT_INT32}}}, in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 1036 case tensorflow::AttrValue::kShape: { in SetOpAttrValueScalar()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 3955 const char kShape[] = "s"; variable
3998 absl::StrAppend(&result->str, kShape);
4053 absl::StrAppend(&result->str, kShape);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 74 constexpr char kShape[] = "Shape"; variable
503 device_cost_impl_.emplace(kShape, in OpLevelCostEstimator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 333 kShape, enumerator
2109 attrs["outfeed_shape"] = {/*required=*/false, AttrTy::kShape, in ParseInstructionRhs()
3802 case AttrTy::kShape: { in ParseAttributeHelper()
|