Searched refs:is_min (Results 1 – 4 of 4) sorted by relevance
34 XlaArgMinMaxOp::XlaArgMinMaxOp(OpKernelConstruction* ctx, bool is_min) in XlaArgMinMaxOp() argument35 : XlaOpKernel(ctx), is_min_(is_min) {} in XlaArgMinMaxOp()
28 explicit XlaArgMinMaxOp(OpKernelConstruction* ctx, bool is_min);
37 DataType output_type, int axis, bool is_min, in ArgMinMax() argument41 if (is_min) { in ArgMinMax()
535 bool is_min = IsMinimumFormat(cpu_desired_num_pixels_); in OnCpuResolutionRequest() local536 if (is_min) { in OnCpuResolutionRequest()