Searched refs:partial_shapes (Results 1 – 4 of 4) sorted by relevance
38 const std::vector<PartialTensorShape>& partial_shapes, const string& name) in PaddingFIFOQueue() argument40 ConvertShapesPartialDimensionsToZero(partial_shapes), name), in PaddingFIFOQueue()41 partial_shapes_(partial_shapes) {} in PaddingFIFOQueue()399 const gtl::ArraySlice<PartialTensorShape>& partial_shapes) { in ConvertShapesPartialDimensionsToZero() argument400 std::vector<TensorShape> shapes(partial_shapes.size()); in ConvertShapesPartialDimensionsToZero()402 const PartialTensorShape& partial = partial_shapes[i]; in ConvertShapesPartialDimensionsToZero()
60 const gtl::ArraySlice<PartialTensorShape>& partial_shapes);
262 std::vector<PartialTensorShape>& partial_shapes = shape_map[node->name()]; in FoldConstants() local264 partial_shapes.resize(ctx->num_outputs()); in FoldConstants()272 &partial_shapes[output_idx])); in FoldConstants()287 recv_partial_shapes.push_back(partial_shapes[it->second]); in FoldConstants()
618 std::vector<PartialTensorShape> partial_shapes(engine_input_shapes.begin(), in GetEngine() local625 partial_shapes, &logger, allocator, calibrator_.get(), &engine, in GetEngine()