Searched refs:shapes_shape_t (Results 1 – 2 of 2) sorted by relevance
796 auto* shapes_shape_t = reinterpret_cast<int64_t*>(parse_output->data.i64); in PrepareParseExample() local797 shapes_shape_t[0] = batch_size; in PrepareParseExample()798 shapes_shape_t[1] = 1; in PrepareParseExample()
1270 auto shapes_shape_t = result->sparse_shapes.back().vec<int64>(); in FastParseExample() local1271 shapes_shape_t(0) = serialized.size(); in FastParseExample()1272 shapes_shape_t(1) = max_num_features; in FastParseExample()1733 auto shapes_shape_t = out_dense_shape->vec<int64>(); in FastParseSingleExample() local1734 shapes_shape_t(0) = num_elements; in FastParseSingleExample()