Searched refs:partial_shapes (Results 1 – 6 of 6) 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()
652 auto& partial_shapes = shape_map[node_name]; in GetGraph() local654 partial_shapes.push_back(inferred_shape.shape); in GetGraph()675 auto& partial_shapes = shape_map[node_name]; in GetGraph() local677 partial_shapes.push_back(inferred_shape.shape); in GetGraph()
80 auto& partial_shapes = (*shape_map)[node_name]; in ConvertGraphShapeInfoToShapeMap() local82 partial_shapes.push_back(inferred_shape.shape); in ConvertGraphShapeInfoToShapeMap()
1068 std::vector<PartialTensorShape> partial_shapes(shapes.begin(), in AllocateCalibrationResources() local1081 partial_shapes, &cache_res->GetLogger(), cache_res->allocator_.get(), in AllocateCalibrationResources()