Searched refs:reshaped_index (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3264 auto reshaped_index = rewriter.create<ReshapeOp>(loc, type, index); in rewriteWithUnknownBegin() local
3267 rewriter.create<TF::LessOp>(loc, reshaped_index, zero); in rewriteWithUnknownBegin()
3271 rewriter.create<TF::AddV2Op>(loc, input_val, reshaped_index); in rewriteWithUnknownBegin()
3273 loc, type, index_negative, wrapped_index, reshaped_index); in rewriteWithUnknownBegin()
|