Searched refs:input_shapes_ (Results 1 – 11 of 11) 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::__anon51f9a25e0111::AddModelBuilder
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | concatenation_tester.h | 55 input_shapes_ = shapes; in InputShapes()
60 return input_shapes_[i]; in InputShape()
63 inline size_t NumInputs() const { return input_shapes_.size(); } in NumInputs()
88 std::vector<std::vector<int32_t>> input_shapes_; variable
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | parse_testdata.cc | 355 if (input_shapes_ == nullptr) { in AddChild()
356 input_shapes_ = new KvMap; in AddChild()
357 Store(input_shapes_); in AddChild()
359 return input_shapes_; in AddChild()
364 if (!input_shapes_) return; in Finish()
365 for (const auto& item : *input_shapes_) { in Finish()
373 KvMap* input_shapes_ = nullptr; member in tflite::testing::Reshape
|
| D | tf_driver.cc | 117 input_shapes_.resize(input_layer.size()); in TfDriver()
122 input_shapes_[i] = Split<int64_t>(input_layer_shape[i], ","); in TfDriver()
159 input_shapes_[id] = Split<int64_t>(csv_values, ","); in ReshapeTensor()
161 CreateTensor(input_types_[id], input_shapes_[id]); in ReshapeTensor()
193 auto tensor = CreateTensor(input_types_[id], input_shapes_[id]); in Invoke()
|
| D | tf_driver.h | 76 std::vector<std::vector<int64_t>> input_shapes_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 210 input_shapes_.push_back(shapes); in AddShape()
259 bool HasShape() const { return !input_shapes_.empty(); } in HasShape()
282 std::vector<std::vector<TensorShape>> input_shapes_;
|
| D | trt_shape_optimization_profiles.cc | 259 if (input_shapes_.size() == 0) { in InitProfiles()
270 for (int i = 0; i < input_shapes_.size(); i++) { in InitProfiles()
271 auto shape_vec = input_shapes_[i]; in InitProfiles()
|
| /external/pytorch/torch/_functorch/ |
| D | autograd_function.py | 399 input_shapes_ = None
414 nonlocal input_shapes_
415 input_shapes_ = tuple(
430 input_shapes = input_shapes_
|
| /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_t>> input_shapes_; variable
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 182 input_shapes_ = input_shapes; in BuildInterpreter()
|
| D | test_util.h | 656 BuildInterpreter(input_shapes_, num_threads, allocate_and_delegate_,
924 std::vector<std::vector<int>> input_shapes_; variable
|