Home
last modified time | relevance | path

Searched refs:x_in_bounds (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dpooling.cc111 auto x_in_bounds = [input_width = ctx.input_shapes[0][2], in GenerateAveragePoolingCode() local
127 !(x_in_bounds(base_x) && in GenerateAveragePoolingCode()
128 x_in_bounds(base_x + output_shape_max_x * attr.strides.w) && in GenerateAveragePoolingCode()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1597 Value x_in_bounds = rewriter.create<AndOp>( in matchAndRewrite() local
1611 rewriter.create<AndOp>(loc, x_in_bounds, y_in_bounds)); in matchAndRewrite()