Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation.cc934 auto converted_shape = in Run() local
936 UpdateLayout(&converted_shape); in Run()
938 HloInstruction::CreateConvert(converted_shape, leaf)); in Run()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc383 auto converted_shape = rewriter.create<IndexCastOp>( in matchAndRewrite() local
391 iota.getLoc(), converted_shape, in matchAndRewrite()