Searched defs:d_shape (Results 1 – 2 of 2) sorted by relevance
507 TensorShape d_shape(n, m, 1, n_gemms); in configure() local
2126 Shape d_shape = ShapeUtil::MakeShape(indices_type, {2}); in OptimizeDotOfGather() local