Home
last modified time | relevance | path

Searched refs:RewriteLayoutWithShardedShape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc139 Status RewriteLayoutWithShardedShape( in RewriteLayoutWithShardedShape() function
207 TF_RETURN_IF_ERROR(RewriteLayoutWithShardedShape( in ReshapeWithCorrectRepresentationAndSharding()
Dxla_helpers.h86 Status RewriteLayoutWithShardedShape(
Dxla_compiler.cc852 TF_RETURN_IF_ERROR(RewriteLayoutWithShardedShape( in XLAShapeForArgument()
881 TF_RETURN_IF_ERROR(RewriteLayoutWithShardedShape( in XLAShapeForArgument()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc121 RewriteLayoutWithShardedShape(arg_sharding, /*use_fast_memory=*/false, in GetXlaInputShapes()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc270 RewriteLayoutWithShardedShape(arg_sharding, /*use_fast_memory=*/false, in GetShardingInfo()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1492 auto status = tensorflow::RewriteLayoutWithShardedShape( in SetEntryTupleShardings()