Searched defs:seed_shape (Results 1 – 4 of 4) sorted by relevance
156 TensorShape seed_shape = ctx->InputShape(1); in Compile() local201 TensorShape seed_shape = ctx->InputShape(1); in Compile() local252 TensorShape seed_shape = ctx->InputShape(1); in Compile() local293 TensorShape seed_shape = ctx->InputShape(1); in Compile() local344 TensorShape seed_shape = ctx->InputShape(1); in Compile() local388 TensorShape seed_shape = ctx->InputShape(1); in Compile() local
174 TensorShape seed_shape = ctx->InputShape(2); in Compile() local
526 TensorShape seed_shape = ctx->InputShape(0); in Compile() local562 TensorShape seed_shape = ctx->InputShape(0); in Compile() local
31 ShapeHandle index_shape, seed_shape, max_index_shape; in StatelessRandomPermuteShape() local