Searched refs:input_shapes_ (Results 1 – 4 of 4) sorted by relevance
| /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()
189 auto tensor = CreateTensor(input_types_[id], input_shapes_[id]); in SetInput()
215 input_shapes_[id] = Split<int64_t>(values_as_string, ","); in ReshapeTensor()
217 CreateTensor(input_types_[id], input_shapes_[id]); in ReshapeTensor()
|
| D | tf_driver.h | 67 std::vector<std::vector<int64_t>> input_shapes_; variable
|
| /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::__anonc3349ced0111::FuseRemoteGraphMultipleAddOpsRewriterTest
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_node.h | 408 for (const auto& s : input_shapes_) { in ToProto()
462 input_shapes_.clear(); in FromProto()
464 auto& shape = input_shapes_[s.first]; in FromProto()
744 std::map<int, std::vector<int64>> input_shapes_; variable
|