Searched refs:ConstantInputReshapedToIntVector (Results 1 – 5 of 5) sorted by relevance
57 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector( in Compile()78 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector( in Compile()
145 Status ConstantInputReshapedToIntVector(int index, std::vector<int64>* out);146 Status ConstantInputReshapedToIntVector(absl::string_view name,
355 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext363 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext
68 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector(1, &axes)); in Compile()
237 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector("dim", &dims)); in Compile()