Searched defs:s_shape (Results 1 – 2 of 2) sorted by relevance
502 TensorShape s_shape = in Compute() local649 TensorShape s_shape = input_tensors[k].shape(); in Compute() local
2369 Shape s_shape = ShapeUtil::MakeShape(indices_type, {1}); in OptimizeDotOfGather() local