Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc3481 for (int64 inserted_dim : dim_numbers.inserted_window_dims()) { in ValidateScatterDimensionNumbers() local
3482 if (inserted_dim < 0 || inserted_dim >= operand_shape.dimensions_size()) { in ValidateScatterDimensionNumbers()
3486 operand_shape.dimensions_size(), inserted_dim); in ValidateScatterDimensionNumbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc5549 auto inserted_dim = expand_dims[0].getSExtValue(); in matchAndRewrite() local
5552 if (inserted_dim < 0) { in matchAndRewrite()
5553 inserted_dim += result_ty.getRank(); in matchAndRewrite()
5555 if (inserted_dim < 0) { in matchAndRewrite()
5560 dims[inserted_dim] = rewriter.create<ConstantIndexOp>(op.getLoc(), 1); in matchAndRewrite()
5568 dims[i >= inserted_dim ? i + 1 : i] = dim; in matchAndRewrite()