HomeSort by relevance Sort by last modified time
    Searched refs:shape_data (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
shape_refiner_test.cc 354 const Tensor* shape_data = c->input_tensor(0);
355 if (shape_data == nullptr) {
360 dims.reserve(shape_data->NumElements());
361 for (int i = 0; i < shape_data->NumElements(); ++i) {
362 dims.emplace_back(c->MakeDim(shape_data->flat<int32>()(i)));
373 const Tensor* shape_data = c->input_tensor(0);
374 if (shape_data == nullptr) {
379 dims.reserve(shape_data->NumElements());
380 for (int i = 0; i < shape_data->NumElements(); ++i) {
381 dims.emplace_back(c->MakeDim(shape_data->flat<int64>()(i)))
448 Node* shape_data; local
482 Node* shape_data; local
515 Node* shape_data; local
541 Node* shape_data; local
567 Node* shape_data; local
598 Node* shape_data; local
628 Node* shape_data; local
648 Node* shape_data; local
669 Node* shape_data; local
690 Node* shape_data; local
711 Node* shape_data; local
733 Node* shape_data; local
753 Node* shape_data; local
788 Node* shape_data; local
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 362 // shape_data is the raw array of ints describing the shape
367 std::vector<int32> shape_data = local
373 for (int i = 0; i < shape_data.size(); i++) {
374 if (shape_data[i] == -1) {
379 product_non_wildcard_dims *= shape_data[i];
388 shape_data[wildcard_index] = input_flat_size / product_non_wildcard_dims;
391 *output_shape.mutable_dims() = shape_data;
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 704 const std::vector<int32> shape_data = { local
706 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph);
745 const std::vector<int32> shape_data = { local
747 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph);
921 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; local
    [all...]

Completed in 3101 milliseconds