Searched defs:gather_op (Results 1 – 5 of 5) sorted by relevance
36 auto gather_op = llvm::cast<mlir::TF::GatherV2Op>(op); in ExpandOp() local170 auto gather_op = llvm::cast<mlir::TF::GatherV2Op>(op); in ComputeLayoutForward() local240 auto gather_op = llvm::cast<mlir::TF::GatherV2Op>(op); in ComputeLayoutBackward() local371 auto gather_op = llvm::cast<mlir::TF::GatherNdOp>(op); in ExpandOp() local470 auto gather_op = llvm::cast<mlir::TF::GatherNdOp>(op); in ComputeLayoutForward() local512 auto gather_op = llvm::cast<mlir::TF::GatherNdOp>(op); in ComputeLayoutBackward() local
132 for (auto* gather_op : gather_ops) { in Run() local178 for (const auto& gather_op : gather_ops) { in Run() local238 for (auto* gather_op : gather_ops) { in Run() local
31 auto* gather_op = model->operators[op_index].get(); in Run() local
1238 tensorflow::NodeDef* gather_op = tensorflow_graph->add_node(); in ConvertGatherOperator() local
2879 mhlo::GatherOp gather_op, OpAdaptor adaptor, in matchAndRewrite()