Searched refs:kShape (Results 1 – 19 of 19) sorted by relevance
/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 | 76 case OperatorType::kShape: in Run()
|
D | quantize.cc | 65 type == OperatorType::kShape || type == OperatorType::kExpandDims || in SupportsQuantization()
|
D | propagate_fixed_sizes.cc | 2241 case OperatorType::kShape: in Run()
|
/external/gemmlowp/public/ |
D | map.h | 87 static const VectorShape kShape = tShape; variable 116 static const VectorShape kShape = tShape; variable
|
/external/tensorflow/tensorflow/core/framework/ |
D | attr_value_util.cc | 218 case AttrValue::kShape: in SummarizeAttrValue() 305 VALIDATE_FIELD(shape, "shape", kShape); in AttrValueHasType()
|
/external/tensorflow/tensorflow/lite/toco/ |
D | tensorflow_util.cc | 102 } else if (attr.second.value_case() == AttrValue::kShape) { in LogDumpGraphDef()
|
D | model.h | 124 kShape, enumerator 1343 TensorFlowShapeOperator() : Operator(OperatorType::kShape) {}
|
D | import_tensorflow.cc | 122 CHECK_EQ(attr.value_case(), AttrValue::kShape); in GetShapeAttr()
|
D | export_tensorflow.cc | 2286 } else if (src_op.type == OperatorType::kShape) { in ConvertOperator()
|
/external/tensorflow/tensorflow/compiler/aot/ |
D | codegen_test_h.golden | 310 static const xla::ProgramShapeProto* kShape = []() { 315 return kShape;
|
/external/gemmlowp/internal/ |
D | output.h | 238 const int pos = VectorType::kShape == VectorShape::Row ? col : row;
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator.cc | 54 constexpr char kShape[] = "Shape"; variable 272 {kShape, wrap(&OpLevelCostEstimator::PredictMetadata)}, in OpLevelCostEstimator()
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tfe_src.cc | 2861 const char kShape[] = "s"; variable 2902 absl::StrAppend(&result->str, kShape); 2949 absl::StrAppend(&result->str, kShape);
|
/external/tensorflow/tensorflow/lite/toco/tflite/ |
D | operator_test.cc | 519 SerializeAndDeserialize(GetOperator("SHAPE", OperatorType::kShape), op); in TEST_F()
|
D | operator.cc | 2444 MakeUnique<Shape>(::tflite::BuiltinOperator_SHAPE, OperatorType::kShape)); in BuildOperatorList()
|
/external/tensorflow/tensorflow/cc/framework/ |
D | cc_op_gen.cc | 220 case AttrValue::kShape: in PrintAttrValue()
|
/external/tensorflow/tensorflow/c/eager/ |
D | c_api.cc | 940 case tensorflow::AttrValue::kShape: { in SetOpAttrValueScalar()
|
/external/tensorflow/tensorflow/c/ |
D | c_api.cc | 1662 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OperationGetAttrMetadata() 1870 if (attr->value_case() != tensorflow::AttrValue::kShape) { in TF_OperationGetAttrTensorShapeProto()
|