Searched refs:shape_2d (Results 1 – 2 of 2) sorted by relevance
328 shape_2d = (int(np.prod(inputs.shape[:split_dim])),330 inputs_2d = np.reshape(inputs, shape_2d)
233 TensorShape shape_2d({dim, dim}); in GrapplerItemFromMetaGraphDef() local246 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef()