Home
last modified time | relevance | path

Searched refs:kShape (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_shape_or_rank.cc28 if (!(op->type == OperatorType::kShape || op->type == OperatorType::kRank)) { in Run()
53 if (op->type == OperatorType::kShape) { in Run()
Dpropagate_array_data_types.cc76 case OperatorType::kShape: in Run()
Dquantize.cc65 type == OperatorType::kShape || type == OperatorType::kExpandDims || in SupportsQuantization()
Dpropagate_fixed_sizes.cc2241 case OperatorType::kShape: in Run()
/external/gemmlowp/public/
Dmap.h87 static const VectorShape kShape = tShape; variable
116 static const VectorShape kShape = tShape; variable
/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.cc218 case AttrValue::kShape: in SummarizeAttrValue()
305 VALIDATE_FIELD(shape, "shape", kShape); in AttrValueHasType()
/external/tensorflow/tensorflow/lite/toco/
Dtensorflow_util.cc102 } else if (attr.second.value_case() == AttrValue::kShape) { in LogDumpGraphDef()
Dmodel.h124 kShape, enumerator
1343 TensorFlowShapeOperator() : Operator(OperatorType::kShape) {}
Dimport_tensorflow.cc122 CHECK_EQ(attr.value_case(), AttrValue::kShape); in GetShapeAttr()
Dexport_tensorflow.cc2286 } else if (src_op.type == OperatorType::kShape) { in ConvertOperator()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen_test_h.golden310 static const xla::ProgramShapeProto* kShape = []() {
315 return kShape;
/external/gemmlowp/internal/
Doutput.h238 const int pos = VectorType::kShape == VectorShape::Row ? col : row;
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc54 constexpr char kShape[] = "Shape"; variable
272 {kShape, wrap(&OpLevelCostEstimator::PredictMetadata)}, in OpLevelCostEstimator()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2861 const char kShape[] = "s"; variable
2902 absl::StrAppend(&result->str, kShape);
2949 absl::StrAppend(&result->str, kShape);
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc519 SerializeAndDeserialize(GetOperator("SHAPE", OperatorType::kShape), op); in TEST_F()
Doperator.cc2444 MakeUnique<Shape>(::tflite::BuiltinOperator_SHAPE, OperatorType::kShape)); in BuildOperatorList()
/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen.cc220 case AttrValue::kShape: in PrintAttrValue()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc940 case tensorflow::AttrValue::kShape: { in SetOpAttrValueScalar()
/external/tensorflow/tensorflow/c/
Dc_api.cc1662 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OperationGetAttrMetadata()
1870 if (attr->value_case() != tensorflow::AttrValue::kShape) { in TF_OperationGetAttrTensorShapeProto()