Searched refs:GetTensorListBufferShape (Results 1 – 4 of 4) sorted by relevance
59 GetTensorListBufferShape(ctx->Input(0), &sum_shape)); in Compile()66 ctx, GetTensorListBufferShape(ctx->Input(i), &operand_shape)); in Compile()
43 Status GetTensorListBufferShape(xla::XlaOp list, xla::Shape* buffer_shape);
308 OP_REQUIRES_OK(ctx, GetTensorListBufferShape(ctx->Input(0), &list_shape)); in Compile()410 OP_REQUIRES_OK(ctx, GetTensorListBufferShape(list, &buffer_xla_shape)); in Compile()
139 Status GetTensorListBufferShape(xla::XlaOp list, xla::Shape* buffer_shape) { in GetTensorListBufferShape() function