Searched refs:TryGetElementShapeFromInput (Results 1 – 1 of 1) sorted by relevance
110 Status TryGetElementShapeFromInput(XlaOpKernelContext* ctx, xla::XlaOp input, in TryGetElementShapeFromInput() function168 TryGetElementShapeFromInput(ctx, element_shape_handle, type, in Compile()240 ctx, TryGetElementShapeFromInput(ctx, element_shape_handle, type, in Compile()