Searched refs:ConstantInputReshapedToIntVector (Results 1 – 4 of 4) sorted by relevance
130 Status ConstantInputReshapedToIntVector(int index, std::vector<int64>* out);131 Status ConstantInputReshapedToIntVector(absl::string_view name,
263 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext271 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext
68 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector(1, &axes)); in Compile()
123 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector("dim", &dims)); in Compile()