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

  /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/
trt_shfn.cc 68 std::vector<DimensionHandle> dim_vec; local
69 dim_vec.emplace_back(context->MakeDim(num_batch));
73 dim_vec.emplace_back(context->MakeDim(dims.d[j]));
78 output_shape = context->MakeShape(dim_vec);
  /external/tensorflow/tensorflow/c/
c_api.cc 664 std::vector<tensorflow::shape_inference::DimensionHandle> dim_vec; local
665 dim_vec.reserve(num_dims);
667 dim_vec.push_back(ic->MakeDim(dims[i]));
669 return ic->MakeShape(dim_vec);
    [all...]

Completed in 115 milliseconds