Searched refs:shape_values (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 139 const std::vector<nvinfer1::Dims>& shape_values, in IncludesShapes()
144 (has_shape_tensor && min.size() != shape_values.size() * 2)) { in IncludesShapes()
147 << shape_values.size(); in IncludesShapes()
171 for (int i = 0; i < shape_values.size(); i++) { in IncludesShapes()
175 auto shape_val = shape_values[i]; in IncludesShapes()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf.cc | 86 std::vector<int32> shape_values( in InferFilterRank() local
89 reinterpret_cast<char*>(shape_values.data())); in InferFilterRank()
90 CHECK_EQ(shape_values.size(), 3); in InferFilterRank()
93 CHECK_EQ(shape_values[2], -1); in InferFilterRank()
94 return shape_values[1]; in InferFilterRank()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 193 DataVec shape_values{{"one", {}}, {"two", shape_tensor}}; in CreateTRTEngine() local
194 TF_CHECK_OK(profile.CollectShapeValues(shape_values)); in CreateTRTEngine()
383 DataVec shape_values{{"one", {}}, {"two", shape_tensor}}; in TEST_P() local
384 TF_CHECK_OK(resource->profiles_.CollectShapeValues(shape_values)); in TEST_P()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to_test.cc | 46 std::initializer_list<ShapeType> shape_values) { in BroadcastToOpModel() argument
49 AddConstInput(GetTensorType<ShapeType>(), shape_values, shape_shape); in BroadcastToOpModel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1084 def _total_size(shape_values): argument
1091 for val in shape_values:
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2835 const auto& shape_values = shape.flat<int32>(); in CreateUpdatedShapeProto() local
2836 if (shape_values(2) != 1 || shape_values(4) != 1) { in CreateUpdatedShapeProto()
2842 new_shape.flat<int32>()(0) = shape_values(0); in CreateUpdatedShapeProto()
2843 new_shape.flat<int32>()(1) = shape_values(1); in CreateUpdatedShapeProto()
2844 new_shape.flat<int32>()(2) = shape_values(3); in CreateUpdatedShapeProto()
2845 new_shape.flat<int32>()(3) = shape_values(5); in CreateUpdatedShapeProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4784 SmallVector<Value, 4> shape_values; in matchAndRewrite() local
4785 shape_values.reserve(input_rank); in matchAndRewrite()
4789 shape_values.push_back(dim_size_value); in matchAndRewrite()
4792 loc, RankedTensorType::get({input_rank}, index_ty), shape_values); in matchAndRewrite()
5570 SmallVector<Value, 4> shape_values; in matchAndRewrite() local
5571 shape_values.reserve(value_rank - 1); in matchAndRewrite()
5582 shape_values.push_back(dim_i); in matchAndRewrite()
5590 shape_values.push_back(dim_i); in matchAndRewrite()
5628 RankedTensorType::get({static_cast<int64_t>(shape_values.size())}, in matchAndRewrite()
5630 shape_values); in matchAndRewrite()
|