Searched refs:new_iota (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 319 auto new_iota = rewriter.create<IotaOp>(iota.getLoc(), iota_type, in matchAndRewrite() local
325 rewriter.replaceOpWithNewOp<BroadcastInDimOp>(iota, result_ty, new_iota, in matchAndRewrite()
406 auto new_iota = rewriter.create<DynamicIotaOp>( in matchAndRewrite() local
414 iota, result_ty, new_iota, iota.output_shape(), broadcast_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3946 HloInstruction* new_iota = computation_->AddInstruction( in HandleReshape() local
3948 iota->SetupDerivedInstruction(new_iota); in HandleReshape()
3954 reshape->shape(), HloOpcode::kMultiply, new_iota, in HandleReshape()
3958 new_reshape = new_iota; in HandleReshape()
|