Searched refs:input_shapes_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/ |
| D | add.cc | 45 for (int8_t i = 0; i < input_shapes_.size(); i++) { in Build()
47 builder, builder.CreateVector<int32_t>(input_shapes_[i].data(), 4), in Build()
55 builder.CreateString(std::to_string(input_shapes_.size())))); in Build()
87 if (input_shapes_.size() <= input) { in SetInputShape()
88 input_shapes_.resize(input + 1); in SetInputShape()
91 input_shapes_[input] = std::move(shape); in SetInputShape()
99 std::vector<std::vector<int32_t>> input_shapes_; member in tflite::__anon5ac0d3af0111::AddModelBuilder
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tf_driver.cc | 116 input_shapes_.resize(input_layer.size()); in TfDriver()
121 input_shapes_[i] = Split<int64_t>(input_layer_shape[i], ","); in TfDriver()
193 auto tensor = CreateTensor(input_types_[id], input_shapes_[id]); in SetInput()
219 input_shapes_[id] = Split<int64_t>(values_as_string, ","); in ReshapeTensor()
221 CreateTensor(input_types_[id], input_shapes_[id]); in ReshapeTensor()
|
| D | tf_driver.h | 69 std::vector<std::vector<int64_t>> input_shapes_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 142 input_shapes_.insert(shapes); in AddShape()
175 bool HasShape() const { return !input_shapes_.empty(); } in HasShape()
184 input_shapes_;
|
| D | trt_shape_optimization_profiles.cc | 62 for (auto& shape_vec : input_shapes_) { in ImplicitBatchModeCompatibleStrategy()
74 for (auto& shape_vec : input_shapes_) { in OptimalStrategy()
102 if (input_shapes_.size() == 0) { in InitProfiles()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform_test.cc | 115 if (!input_shapes_.empty()) { in FuseInternal()
118 {input_shapes_}})); in FuseInternal()
168 input_shapes_ = "1,1,1,1"; in SetInputShapeType()
207 string input_shapes_; member in tensorflow::graph_transforms::__anon7e4ad84b0111::FuseRemoteGraphMultipleAddOpsRewriterTest
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 407 for (const auto& s : input_shapes_) { in ToProto()
461 input_shapes_.clear(); in FromProto()
463 auto& shape = input_shapes_[s.first]; in FromProto()
743 std::map<int, std::vector<int64>> input_shapes_; variable
|