Searched refs:kRank (Results 1 – 10 of 10) 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()
56 } else if (op->type == OperatorType::kRank) { in Run()
|
| D | propagate_array_data_types.cc | 74 case OperatorType::kRank: in Run()
|
| D | propagate_fixed_sizes.cc | 2321 case OperatorType::kRank: in Run()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 260 {{OperatorType::kRank, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
|
| D | operator_test.cc | 158 CheckSimpleOperator<TensorFlowRankOperator>("RANK", OperatorType::kRank); in TEST_F()
|
| D | operator.cc | 2078 ::tflite::BuiltinOperator_RANK, OperatorType::kRank)); in BuildOperatorList()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 74 kRank, enumerator
1334 TensorFlowRankOperator() : Operator(OperatorType::kRank) {}
|
| D | export_tensorflow.cc | 2331 } else if (src_op.type == OperatorType::kRank) { in ConvertOperator()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 69 constexpr int kRank = 4; variable
1722 if (shape.dim_size() != kRank) { in IsInputConvertible()
1774 IsAlongAxis(*squeeze_dims_attr, indices({'H', 'W'}), kRank)) || in IsDimsSupported()
1776 IsAlongAxis(*squeeze_dims_attr, indices({'N', 'H', 'W'}), kRank)); in IsDimsSupported()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 72 constexpr char kRank[] = "Rank"; variable
501 device_cost_impl_.emplace(kRank, in OpLevelCostEstimator()
|