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