Searched refs:ConstantInputReshapedToIntVector (Results 1 – 6 of 6) sorted by relevance
57 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector( in Compile()78 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector( in Compile()
180 Status ConstantInputReshapedToIntVector(183 Status ConstantInputReshapedToIntVector(
412 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext420 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext
68 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector(1, &axes)); in Compile()
117 ctx->ConstantInputReshapedToIntVector(1, &partitions_static).ok(); in Compile()
265 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector("dim", &dims)); in Compile()