Home
last modified time | relevance | path

Searched defs:tile_op (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dmeta_spmd_expander.cc363 auto tile_op = llvm::cast<mlir::TF::TileOp>(op); in ExpandOp() local
459 auto tile_op = llvm::cast<mlir::TF::TileOp>(op); in ComputeLayoutForward() local
504 auto tile_op = llvm::cast<mlir::TF::TileOp>(op); in ComputeLayoutBackward() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dconvert_trivial_tile_to_concat.cc32 auto* tile_op = static_cast<TransposeOperator*>(tile_it->get()); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc1931 tensorflow::NodeDef* tile_op = tensorflow_graph->add_node(); in ConvertTileOperator() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer_test.cc3010 auto tile_op = ops::Tile(scope.WithOpName("tile").WithDevice("/device:GPU:0"), in TEST_F() local